1 #pragma once 2 3 #include <torch/csrc/python_headers.h> 4 5 #include <c10/core/QScheme.h> 6 7 #include <string> 8 9 constexpr int QSCHEME_NAME_LEN = 64; 10 11 struct THPQScheme { 12 PyObject_HEAD at::QScheme qscheme; 13 // NOLINTNEXTLINE(cppcoreguidelines-avoid-c-arrays,modernize-avoid-c-arrays) 14 char name[QSCHEME_NAME_LEN + 1]; 15 }; 16 17 extern PyTypeObject THPQSchemeType; 18 THPQScheme_Check(PyObject * obj)19inline bool THPQScheme_Check(PyObject* obj) { 20 return Py_TYPE(obj) == &THPQSchemeType; 21 } 22 23 PyObject* THPQScheme_New(at::QScheme qscheme, const std::string& name); 24 25 void THPQScheme_init(PyObject* module); 26