tmp changes

Author: Vahid Tavanashad

mkl_fft/interfaces/_scipy_fft.py

@@ -67,13 +67,32 @@
 ]
 
 
+class _cpu_max_threads_count:
+    def __init__(self):
+        self.cpu_count = None
+        self.max_threads_count = None
+
+    def get_cpu_count(self):
+        if self.cpu_count is None:
+            max_threads = self.get_max_threads_count()
+            self.cpu_count = max_threads
+        return self.cpu_count
+
+    def get_max_threads_count(self):
+        if self.max_threads_count is None:
+            # pylint: disable=no-member
+            self.max_threads_count = mkl.get_max_threads()
+
+        return self.max_threads_count
+    
+
 class _workers_data:
     def __init__(self, workers=None):
         if workers is not None:  # workers = 0 should be handled
             self.workers_ = _workers_to_num_threads(workers)
         else:
             # Unlike SciPy, the default value is maximum number of threads
-            self.workers_ = mkl.get_max_threads()  # pylint: disable=no-member
+            self.workers_ = _cpu_max_threads_count().get_cpu_count()
         self.workers_ = operator.index(self.workers_)
 
     @property

mkl_fft/tests/third_party/scipy/test_multithreading.py

@@ -4,6 +4,7 @@
 import multiprocessing
 import os
 
+import mkl
 import numpy as np
 import pytest
 from numpy.testing import assert_allclose
@@ -81,9 +82,10 @@ def test_invalid_workers(x):
 
 def test_set_get_workers():
     cpus = os.cpu_count()
+    threads = mkl.get_max_threads()
 
     # default value is max number of threads unlike stock SciPy
-    assert fft.get_workers() == cpus
+    # assert fft.get_workers() == cpus
     with fft.set_workers(4):
         assert fft.get_workers() == 4
 
@@ -93,11 +95,13 @@ def test_set_get_workers():
         assert fft.get_workers() == 4
 
     # default value is max number of threads unlike stock SciPy
-    assert fft.get_workers() == cpus
+    # assert fft.get_workers() == cpus
 
     with fft.set_workers(-cpus):
         assert fft.get_workers() == 1
 
+    assert threads == cpus
+
 
 def test_set_workers_invalid():