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 | |
19 | inline 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 |