safeguards

jeremiedbb · jeremiedbb · commit b104d5e7bb25 · 2024-07-03T16:49:25.000+02:00
diff --git a/sklearn/utils/_openmp_helpers.pyx b/sklearn/utils/_openmp_helpers.pyx
@@ -111,14 +111,20 @@ def set_openblas_openmp_callback():
     cdef openblas_set_threads_callback_function_type f_ptr
 
     for ct in openblas_controllers:
+        if not hasattr(ct, "dynlib"):
+            # too old version of threadpoolctl
+            continue
+
         lib = ct.dynlib
 
-        # openblas_set_threads_callback_function is available since v0.3.28
-        if hasattr(lib, "openblas_set_threads_callback_function"):
-            func = lib.openblas_set_threads_callback_function
+        if not hasattr(lib, "openblas_set_threads_callback_function"):
+            # openblas_set_threads_callback_function is available since v0.3.28
+            continue
+
+        func = lib.openblas_set_threads_callback_function
 
-            # cast to the correct function pointer type
-            f_ptr = (<openblas_set_threads_callback_function_type*><size_t>addressof(func))[0]
-            f_ptr(openblas_openmp_callback)
+        # cast to the correct function pointer type
+        f_ptr = (<openblas_set_threads_callback_function_type*><size_t>addressof(func))[0]
+        f_ptr(openblas_openmp_callback)
 
-            print("OpenBLAS OpenMP backend callback set")
+        print("OpenBLAS OpenMP backend callback set")
