Updating cuquantum version

pyproject.toml

@@ -56,7 +56,7 @@ ipython = "^8.10.0"
 qulacs = { version = "^0.6.4", markers = "(sys_platform == 'darwin' and python_version > '3.9') or sys_platform != 'darwin'" }
 seaborn = "^0.13.2"
 ipykernel = "^6.29.4"
-qibojit = { git = "https://github.com/qiboteam/qibojit.git", branch="cupy_matrices" }
+qibojit = { git = "https://github.com/qiboteam/qibojit.git", branch="update_cuquantum" }
 
 [tool.poetry.group.tests]
 optional = true
@@ -70,7 +70,7 @@ pytest-cov = "^4.0.0"
 pylint = "3.1.0"
 matplotlib = "^3.7.0"
 torch = "^2.1.1,<2.4"
-qibojit = { git = "https://github.com/qiboteam/qibojit.git", branch="cupy_matrices" }
+qibojit = { git = "https://github.com/qiboteam/qibojit.git", branch="update_cuquantum" }
 qibotn = { git = "https://github.com/qiboteam/qibotn.git" }
 qiboml = { git  = "https://github.com/qiboteam/qiboml.git" }
 stim = "^1.12.0"
@@ -91,7 +91,7 @@ optional = true
 [tool.poetry.group.cuda11.dependencies]
 cupy-cuda11x = "^13.1.0"
 cuquantum-python-cu11 = "^23.3.0"
-qibojit = { git = "https://github.com/qiboteam/qibojit.git", branch="cupy_matrices" }
+qibojit = { git = "https://github.com/qiboteam/qibojit.git", branch="update_cuquantum" }
 qibotn = { git = "https://github.com/qiboteam/qibotn.git" }
 
 
@@ -101,7 +101,7 @@ optional = true
 [tool.poetry.group.cuda12.dependencies]
 cupy-cuda12x = "^13.1.0"
 cuquantum-python-cu12 = "^23.3.0"
-qibojit = { git = "https://github.com/qiboteam/qibojit.git", branch="cupy_matrices" }
+qibojit = { git = "https://github.com/qiboteam/qibojit.git", branch="update_cuquantum" }
 qibotn = { git = "https://github.com/qiboteam/qibotn.git" }
 
 [tool.poetry.extras]

src/qibo/backends/npmatrices.py

@@ -15,7 +15,9 @@ def __init__(self, dtype):
         self.np = np
 
     def _cast(self, x, dtype):
-        return self.np.array(x, dtype=dtype)
+        if isinstance(x, list):
+            return self.np.array(x, dtype=dtype)
+        return x.astype(dtype)
 
     @cached_property
     def H(self):

src/qibo/transpiler/unitary_decompositions.py

@@ -32,8 +32,8 @@ def u3_decomposition(unitary, backend):
     minus = backend.np.angle(su2[1, 0])
     phi = plus + minus
     lam = plus - minus
-
-    return theta, phi, lam
+    # explicit conversion to float to avoid issue on GPU
+    return float(theta), float(phi), float(lam)
 
 
 def calculate_psi(unitary, backend, magic_basis=magic_basis):
@@ -51,7 +51,6 @@ def calculate_psi(unitary, backend, magic_basis=magic_basis):
     Returns:
         ndarray: Eigenvectors in the computational basis and eigenvalues of :math:`U^{T} U`.
     """
-
     magic_basis = backend.cast(magic_basis)
     unitary = backend.cast(unitary)
     # write unitary in magic basis

tests/test_backends.py

@@ -7,6 +7,8 @@
 from qibo import construct_backend, gates, list_available_backends, set_backend
 from qibo.backends import MetaBackend
 
+from .conftest import AVAILABLE_BACKENDS
+
 ####################### Test `matrix` #######################
 GATES = [
     ("H", (0,), np.array([[1, 1], [1, -1]]) / np.sqrt(2)),
@@ -116,26 +118,12 @@ def test_list_available_backends():
     qulacs = (
         False if platform.system() == "Darwin" and sys.version_info[1] == 9 else True
     )
-    try:
-        import cupy
-
-        cupy_available = True
-    except ModuleNotFoundError:
-        cupy_available = False
-    try:
-        import cuquantum
-
-        cuquantum_available = True
-    except ModuleNotFoundError:
-        cuquantum_available = False
-
     available_backends = {
         "numpy": True,
         "qulacs": qulacs,
         "qibojit": {
-            "numba": True,
-            "cupy": cupy_available,
-            "cuquantum": cuquantum_available,
+            platform: any(platform in backend for backend in AVAILABLE_BACKENDS)
+            for platform in ["numba", "cupy", "cuquantum"]
         },
         "qibolab": False,
         "qibo-cloud-backends": False,

tests/test_transpiler_unitary_decompositions.py

@@ -72,7 +72,6 @@ def test_eigenbasis_entanglement(backend, seed):
 def test_v_decomposition(backend, seed):
     """Check that V_A V_B |psi_k> = |phi_k> according to Lemma 1."""
     unitary = random_unitary(4, seed=seed, backend=backend)
-
     psi, _ = calculate_psi(unitary, backend=backend)
     va, vb = calculate_single_qubit_unitaries(psi, backend=backend)
     assert_single_qubits(backend, psi, va, vb)