fix: rolling back

qiboteam · Jan 24, 2025 · 36844fa · 36844fa
1 parent 4e4e611
commit 36844fa
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 17 deletions.
diff --git a/src/qibojit/backends/clifford_operations_cpu.py b/src/qibojit/backends/clifford_operations_cpu.py
@@ -1,9 +1,8 @@
 """Set of custom Numba operations for the Clifford backend."""
 
 import numpy as np
-from numba import njit, prange, set_num_threads, uint64
+from numba import njit, prange, uint64
 
-set_num_threads(1)
 PARALLEL = True
 
 

diff --git a/src/qibojit/backends/cpu.py b/src/qibojit/backends/cpu.py
@@ -1,4 +1,3 @@
-import os
 import sys
 
 import numpy as np
@@ -26,15 +25,6 @@
     "GeneralizedfSim": "apply_fsim",
 }
 
-if os.environ.get("NUMBA_NUM_THREADS") is None:
-    NTHREADS = (
-        psutil.cpu_count(logical=False)
-        if sys.platform == "darwin"
-        else len(psutil.Process().cpu_affinity())
-    )
-
-    os.environ["NUMBA_NUM_THREADS"] = str(NTHREADS)
-
 
 class NumbaBackend(NumpyBackend):
     def __init__(self):

diff --git a/src/qibojit/tests/test_backends.py b/src/qibojit/tests/test_backends.py
@@ -15,12 +15,10 @@ def test_device_setter(backend):
 
 
 def test_thread_setter(backend):
-    # import numba
+    import numba
 
-    # original_threads = numba.get_num_threads()
-    with pytest.raises(RuntimeError):
-        backend.set_threads(1)
-    # assert numba.get_num_threads() == 1
+    original_threads = numba.get_num_threads()
+    assert numba.get_num_threads() == 1
 
 
 @pytest.mark.parametrize("array_type", [None, "float32", "float64"])