feat: implemented I() for cupy matrices

qiboteam · Feb 6, 2025 · ce4f083 · ce4f083
1 parent cb89902
commit ce4f083
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/qibojit/backends/matrices.py b/src/qibojit/backends/matrices.py
@@ -75,6 +75,9 @@ def __init__(self, dtype):
 
         self.cp = cp
 
+    def I(self, n=2):
+        return self.cp.eye(n, dtype=self.dtype)
+
     def _cast(self, x, dtype):
         is_cupy = [
             isinstance(item, self.cp.ndarray) for sublist in x for item in sublist