From 36844fa194940890b377c384a1e44370fedffa85 Mon Sep 17 00:00:00 2001 From: BrunoLiegiBastonLiegi Date: Fri, 24 Jan 2025 11:07:17 +0100 Subject: [PATCH] fix: rolling back --- src/qibojit/backends/clifford_operations_cpu.py | 3 +-- src/qibojit/backends/cpu.py | 10 ---------- src/qibojit/tests/test_backends.py | 8 +++----- 3 files changed, 4 insertions(+), 17 deletions(-) diff --git a/src/qibojit/backends/clifford_operations_cpu.py b/src/qibojit/backends/clifford_operations_cpu.py index 22ec293..581d939 100644 --- 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 index 195ccb2..97c7062 100644 --- 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 index e28bd0c..5193f07 100644 --- 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"])