Merge pull request #231 from thinking-tower/bugfix/segfault-after-imp…

…ort-kp-and-exit Set kp_debug, kp_info, kp_warning and kp_error to py::none() when the program terminates.
KomputeProject · Jun 12, 2021 · e77a684 · e77a684
2 parents 6ffc934 + 0f84be9
commit e77a684
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/python/src/main.cpp b/python/src/main.cpp
@@ -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;