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 | |
18 | inline bool THPFInfo_Check(PyObject* obj) { |
19 | return Py_TYPE(obj) == &THPFInfoType; |
20 | } |
21 | |
22 | inline bool THPIInfo_Check(PyObject* obj) { |
23 | return Py_TYPE(obj) == &THPIInfoType; |
24 | } |
25 | |
26 | void THPDTypeInfo_init(PyObject* module); |
27 |