telling quimb to use same threads as qibojit

qiboteam · Mar 1, 2024 · 77c0475 · 77c0475
1 parent 6e42375
commit 77c0475
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/qibo/backends/__init__.py b/src/qibo/backends/__init__.py
@@ -9,7 +9,6 @@
 
 
 def construct_backend(backend, **kwargs):
-    print(backend)
     if backend == "qibojit":
         from qibojit.backends import CupyBackend, CuQuantumBackend, NumbaBackend
 

diff --git a/tests/test_backend_qibotn.py b/tests/test_backend_qibotn.py
@@ -1,6 +1,11 @@
-from qibotn.backends.quimb import QuimbBackend
+import os
 
 import qibo
+
+# Force quimb to use qibojit default number of threads.
+os.environ["NUMBA_NUM_THREADS"] = f"{qibo.get_threads()}"
+from qibotn.backends.quimb import QuimbBackend
+
 from qibo.backends import GlobalBackend