Updating cuquantum version

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)