Skip to content

Commit 79cf4fc

Browse files
committed
add pybind11::call_guard<pybind11::gil_scoped_release>()
Signed-off-by: yuanyuxing.yyx <[email protected]>
1 parent e6a7e02 commit 79cf4fc

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

fastsafetensors/cpp/ext.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -745,16 +745,16 @@ PYBIND11_MODULE(__MOD_NAME__, m)
745745

746746
pybind11::class_<nogds_file_reader>(m, "nogds_file_reader")
747747
.def(pybind11::init<const bool, const uint64_t, const int, bool>())
748-
.def("submit_read", &nogds_file_reader::submit_read)
749-
.def("wait_read", &nogds_file_reader::wait_read);
748+
.def("submit_read", &nogds_file_reader::submit_read, pybind11::call_guard<pybind11::gil_scoped_release>())
749+
.def("wait_read", &nogds_file_reader::wait_read, pybind11::call_guard<pybind11::gil_scoped_release>());
750750

751751
pybind11::class_<gds_file_handle>(m, "gds_file_handle")
752752
.def(pybind11::init<std::string, bool, bool>());
753753

754754
pybind11::class_<gds_file_reader>(m, "gds_file_reader")
755755
.def(pybind11::init<const int, bool>())
756-
.def("submit_read", &gds_file_reader::submit_read)
757-
.def("wait_read", &gds_file_reader::wait_read);
756+
.def("submit_read", &gds_file_reader::submit_read, pybind11::call_guard<pybind11::gil_scoped_release>())
757+
.def("wait_read", &gds_file_reader::wait_read, pybind11::call_guard<pybind11::gil_scoped_release>());
758758

759759
pybind11::class_<cpp_metrics_t>(m, "cpp_metrics")
760760
.def(pybind11::init<>())

0 commit comments

Comments
 (0)