1 #pragma once 2 3 #include <torch/csrc/python_headers.h> 4 5 #include <ATen/ATen.h> 6 7 struct THPDTypeInfo { 8 PyObject_HEAD at::ScalarType type; 9 }; 10 11 struct THPFInfo : THPDTypeInfo {}; 12 13 struct THPIInfo : THPDTypeInfo {}; 14 15 extern PyTypeObject THPFInfoType; 16 extern PyTypeObject THPIInfoType; 17 THPFInfo_Check(PyObject * obj)18inline bool THPFInfo_Check(PyObject* obj) { 19 return Py_TYPE(obj) == &THPFInfoType; 20 } 21 THPIInfo_Check(PyObject * obj)22inline bool THPIInfo_Check(PyObject* obj) { 23 return Py_TYPE(obj) == &THPIInfoType; 24 } 25 26 void THPDTypeInfo_init(PyObject* module); 27