Function PyObject_ToInt64

paddle/fluid/pybind/op_function_common.cc:156–177

}

int64_t PyObject_ToInt64(PyObject* obj) {
  int64_t res = 0;
  if ((PyLong_Check(obj) && !PyBool_Check(obj)) ||  // NOLINT
      PyObject_CheckVarType(obj) ||                 // NOLINT
      PyObject_CheckDataType(obj) ||                // NOLINT
      (PyObject_CheckTensor(obj) &&
       reinterpret_cast<TensorObject*>(obj)->tensor.numel() == 1)) {
    res = PyLong_AsLongLong(obj);
    return res;
  }
  std::string type_name =
      std::string(reinterpret_cast<PyTypeObject*>(obj->ob_type)->tp_name);
  if (type_name.find("numpy.int") != std::string::npos) {
    auto num_obj = PyNumber_Long(obj);
    res = PyLong_AsLongLong(num_obj);
    Py_DECREF(num_obj);
  } else {
    PADDLE_THROW(common::errors::InvalidType("Cannot convert %s to long long",
                                             type_name));
  }
  return res;
}

uint64_t PyObject_ToUInt64(PyObject* obj) {
  uint64_t res = 0;