fix: using cupy's expm

qiboteam · Feb 24, 2025 · eac2ed7 · eac2ed7
1 parent 2be96c3
commit eac2ed7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/qibojit/backends/gpu.py b/src/qibojit/backends/gpu.py
@@ -108,6 +108,10 @@ def kernel_loader(name, ktype):
         # set the engine of the quantum info operators
         self.qinfo.ENGINE = cp
 
+        from cupyx.scipy.linalg import expm
+
+        self.qinfo.expm = expm
+
     def set_precision(self, precision):
         super().set_precision(precision)
         if self.dtype == "complex128":