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