thek0tyara/llama-cpp-turboquant
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 4

Set kp_debug, kp_info, kp_warning and kp_error to py::none() when the program terminates.

Browse source
This commit is contained in:
Darius Tan 2021-06-12 21:50:37 +12:00
parent 6ffc934aad
commit 0f84be9d5b
1 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
python/src/main.cpp
View file

@ -235,6 +235,13 @@ PYBIND11_MODULE(kp, m) {
return kp::py::vkPropertiesToDict(properties);
}, "Return a dict containing information about the device");
auto atexit = py::module_::import("atexit");
atexit.attr("register")(py::cpp_function([](){
kp_debug = py::none();
kp_info = py::none();
kp_warning = py::none();
kp_error = py::none();
}));
#ifdef VERSION_INFO
m.attr("__version__") = VERSION_INFO;