mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
067d203b22
This is a modified version of https://github.com/pytorch/pytorch/pull/130341 that preserve support for older pybind version. Pull Request resolved: https://github.com/pytorch/pytorch/pull/136370 Approved by: https://github.com/Skylion007, https://github.com/malfet
57 lines
1.5 KiB
C++
57 lines
1.5 KiB
C++
#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
|