1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
#include <torch/csrc/utils/python_symnode.h>
namespace torch {
py::handle get_symint_class() {
// NB: leak
#if IS_PYBIND_2_13_PLUS
PYBIND11_CONSTINIT static py::gil_safe_call_once_and_store<py::object>
storage;
return storage
.call_once_and_store_result([]() -> py::object {
return py::module::import("torch").attr("SymInt");
})
.get_stored();
#else
static py::handle symint_class =
py::object(py::module::import("torch").attr("SymInt")).release();
return symint_class;
#endif
}
py::handle get_symfloat_class() {
// NB: leak
#if IS_PYBIND_2_13_PLUS
PYBIND11_CONSTINIT static py::gil_safe_call_once_and_store<py::object>
storage;
return storage
.call_once_and_store_result([]() -> py::object {
return py::module::import("torch").attr("SymFloat");
})
.get_stored();
#else
static py::handle symfloat_class =
py::object(py::module::import("torch").attr("SymFloat")).release();
return symfloat_class;
#endif
}
py::handle get_symbool_class() {
// NB: leak
#if IS_PYBIND_2_13_PLUS
PYBIND11_CONSTINIT static py::gil_safe_call_once_and_store<py::object>
storage;
return storage
.call_once_and_store_result([]() -> py::object {
return py::module::import("torch").attr("SymBool");
})
.get_stored();
#else
static py::handle symbool_class =
py::object(py::module::import("torch").attr("SymBool")).release();
return symbool_class;
#endif
}
} // namespace torch
|
