fix: fixed patching problem

qiboteam · Jan 9, 2024 · bb7cc10 · bb7cc10
1 parent aaf19ff
commit bb7cc10
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/qibojit/backends/cpu.py b/src/qibojit/backends/cpu.py
@@ -1,3 +1,5 @@
+from importlib.util import find_spec, module_from_spec
+
 import numpy as np
 from numba import njit
 from qibo.backends import clifford_operations
@@ -72,7 +74,9 @@ def __init__(self):
         else:
             self.set_threads(len(psutil.Process().cpu_affinity()))
 
-        self.clifford_operations = clifford_operations
+        spec = find_spec("qibo.backends.clifford_operations")
+        self.clifford_operations = module_from_spec(spec)
+        spec.loader.exec_module(self.clifford_operations)
         for method in dir(clifford_operations_cpu):
             setattr(
                 self.clifford_operations,

diff --git a/src/qibojit/backends/gpu.py b/src/qibojit/backends/gpu.py
@@ -1,3 +1,5 @@
+from importlib.util import find_spec, module_from_spec
+
 import numpy as np
 from qibo.backends import clifford_operations
 from qibo.backends.numpy import NumpyBackend
@@ -100,7 +102,9 @@ def kernel_loader(name, ktype):
         # number of available GPUs (for multigpu)
         self.ngpus = cp.cuda.runtime.getDeviceCount()
 
-        self.clifford_operations = clifford_operations
+        spec = find_spec("qibo.backends.clifford_operations")
+        self.clifford_operations = module_from_spec(spec)
+        spec.loader.exec_module(self.clifford_operations)
         for method in dir(clifford_operations_gpu):
             setattr(
                 self.clifford_operations,