Function tensor__div__method

paddle/fluid/pybind/eager_math_op_patch.cc:709–818

  return ToPyObject(ret);

  EAGER_CATCH_AND_THROW_RETURN_NULL
}

static PyObject* tensor__div__method(TensorObject* self,
                                     PyObject* args,
                                     PyObject* kwargs) {
  phi::RecordEvent pythonc_record_event(
      "__div__ pybind_patch_func", phi::TracerEventType::UserDefined, 1);

  EAGER_TRY

  VLOG(6) << "Running Eager tensor__div__method";

  SetPythonStack();

  // Set Device ID
  auto place = egr::Controller::Instance().GetExpectedPlace();
  SetDevice(place);

  Tensor ret;

  Tensor self_tensor = self->tensor;
  PyObject* other_obj = PyTuple_GET_ITEM(args, 0);

  // 1. scalar exists cases
  if (PyFloat_Check(other_obj) || PyCheckInteger(other_obj) ||
      IsNumpyType(other_obj)) {
    if (_supported_int_dtype_.find(self_tensor.dtype()) !=
        _supported_int_dtype_.end()) {
      eager_gil_scoped_release guard;
      self_tensor = cast_ad_func(self_tensor, DataType::FLOAT32);
    }

    double other = CastPyArg2Double(other_obj, "__div__", 0);
    {
      eager_gil_scoped_release guard;
      ret = CallScalarFunction(self_tensor, other, "div");
    }
    return ToPyObject(ret);
  } else if (PyComplex_Check(other_obj)) {
    if (is_support_complex(self_tensor.dtype()) == false) {
      eager_gil_scoped_release guard;
      self_tensor = cast_ad_func(
          self_tensor, promoteTypes(self_tensor.dtype(), DataType::COMPLEX64));
    }
  }

  // 2. create or get tensor for other_obj
  Tensor other_tensor;
  if (PyCheckTensor(other_obj)) {
    auto& self_tensor_ref_addr = self->tensor;
    auto& other_tensor_ref_addr = CastPyArg2Tensor(other_obj, 0);
    const phi::distributed::ProcessMesh* mesh = nullptr;
    if (InputsContainDistTensor(
            &mesh, self_tensor_ref_addr, other_tensor_ref_addr)) {
      ConvertAllInputsToDistTensor(
          mesh, self_tensor_ref_addr, other_tensor_ref_addr);
    }