| 239 | } |
| 240 | |
| 241 | double PyObject_ToDouble(PyObject* obj) { |
| 242 | double res = 0.0; |
| 243 | if (PyFloat_Check(obj) || PyLong_Check(obj) || |
| 244 | (PyObject_CheckTensor(obj) && |
| 245 | reinterpret_cast<TensorObject*>(obj)->tensor.numel() == 1)) { |
| 246 | res = PyFloat_AsDouble(obj); |
| 247 | return res; |
| 248 | } |
| 249 | auto type_name = |
| 250 | std::string(reinterpret_cast<PyTypeObject*>(obj->ob_type)->tp_name); |
| 251 | if (type_name.find("numpy") != std::string::npos && |
| 252 | type_name.find("numpy.complex") == std::string::npos) { |
| 253 | auto num_obj = PyNumber_Float(obj); |
| 254 | res = PyFloat_AsDouble(num_obj); |
| 255 | Py_DECREF(num_obj); |
| 256 | } else { |
| 257 | PADDLE_THROW( |
| 258 | common::errors::InvalidType("Cannot convert %s to double", type_name)); |
| 259 | } |
| 260 | return res; |
| 261 | } |
| 262 | |
| 263 | bool PyObject_CheckComplexOrToComplex(PyObject** obj) { |
| 264 | if (PyComplex_Check(*obj) || |
no test coverage detected