#pragma once
|
|
#include <ATen/ATen.h>
|
#include <torch/csrc/python_headers.h>
|
#include <torch/csrc/WindowsTorchApiMacro.h>
|
|
const int DTYPE_NAME_LEN = 64;
|
|
struct TORCH_API THPDtype {
|
PyObject_HEAD
|
at::ScalarType scalar_type;
|
char name[DTYPE_NAME_LEN + 1];
|
};
|
|
TORCH_API extern PyTypeObject THPDtypeType;
|
|
inline bool THPDtype_Check(PyObject *obj) {
|
return Py_TYPE(obj) == &THPDtypeType;
|
}
|
|
inline bool THPPythonScalarType_Check(PyObject *obj) {
|
return obj == (PyObject*)(&PyFloat_Type) ||
|
obj == (PyObject*)(&PyBool_Type) ||
|
#if PY_MAJOR_VERSION == 2
|
obj == (PyObject*)(&PyInt_Type) ||
|
#endif
|
obj == (PyObject*)(&PyLong_Type);
|
}
|
|
PyObject * THPDtype_New(at::ScalarType scalar_type, const std::string& name);
|
|
void THPDtype_init(PyObject *module);
|
