xref: /aosp_15_r20/external/pytorch/torch/csrc/TypeInfo.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
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)18 inline bool THPFInfo_Check(PyObject* obj) {
19   return Py_TYPE(obj) == &THPFInfoType;
20 }
21 
THPIInfo_Check(PyObject * obj)22 inline bool THPIInfo_Check(PyObject* obj) {
23   return Py_TYPE(obj) == &THPIInfoType;
24 }
25 
26 void THPDTypeInfo_init(PyObject* module);
27