add precision arg

qiboteam · Oct 4, 2024 · 4a81d92 · 4a81d92
1 parent 51b54c2
commit 4a81d92
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/qibojit/backends/gpu.py b/src/qibojit/backends/gpu.py
@@ -539,12 +539,15 @@ def calculate_matrix_exp(self, a, matrix, eigenvectors=None, eigenvalues=None):
             ud = self.cp.transpose(self.cp.conj(eigenvectors))
             return self.cp.matmul(eigenvectors, self.cp.matmul(expd, ud))
 
-    def calculate_matrix_power(self, matrix, power: Union[float, int]):
+    def calculate_matrix_power(
+        self, matrix, power: Union[float, int], precision_singularity: float = 1e-14
+    ):
+
         if isinstance(power, int):
             return self.cp.linalg.matrix_power(matrix, power)
 
         copied = self.to_numpy(matrix)
-        copied = super().calculate_matrix_power(copied, power)
+        copied = super().calculate_matrix_power(copied, power, precision_singularity)
 
         return self.cast(copied, dtype=copied.dtype)