Stop utilisation polling before cleaning up

mark9064 · Dec 23, 2019 · c65cb4e · c65cb4e
1 parent adb0e79
commit c65cb4e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pyamdgpuinfo/pyamdgpuinfo.pyx b/pyamdgpuinfo/pyamdgpuinfo.pyx
@@ -306,6 +306,8 @@ cpdef object cleanup():
         but there's no point running it when your program finishes as the
         resources will be automatically freed when the process ends.
     """
+    if "thread_args" in GLOBALS:
+        stop_utilisation_polling()
     GLOBALS.pop("py_gpus", None)
     devices_pointer = GLOBALS.pop("gpus", None)
     if devices_pointer is not None: