diff --git a/src/qibotn/backends/cutensornet.py b/src/qibotn/backends/cutensornet.py index 4f1c409..a813a7e 100644 --- a/src/qibotn/backends/cutensornet.py +++ b/src/qibotn/backends/cutensornet.py @@ -16,7 +16,7 @@ class CuTensorNet(QibotnBackend, NumpyBackend): # pragma: no cover super().__init__() import cuquantum from cuquantum import cutensornet as cutn # pylint: disable=import-error - + if runcard is not None: self.MPI_enabled = runcard.get("MPI_enabled", False) self.NCCL_enabled = runcard.get("NCCL_enabled", False) diff --git a/src/qibotn/eval.py b/src/qibotn/eval.py index 2d7c441..9d34275 100644 --- a/src/qibotn/eval.py +++ b/src/qibotn/eval.py @@ -62,9 +62,9 @@ def dense_vector_tn_MPI(qibo_circ, datatype, n_samples=8): Dense vector of quantum circuit. """ + import cuquantum.cutensornet as cutn from cuquantum import Network from mpi4py import MPI - import cuquantum.cutensornet as cutn root = 0 comm = MPI.COMM_WORLD @@ -150,10 +150,10 @@ def dense_vector_tn_nccl(qibo_circ, datatype, n_samples=8): Returns: Dense vector of quantum circuit. """ + import cuquantum.cutensornet as cutn from cupy.cuda import nccl from cuquantum import Network from mpi4py import MPI - import cuquantum.cutensornet as cutn root = 0 comm_mpi = MPI.COMM_WORLD @@ -259,10 +259,10 @@ def expectation_pauli_tn_nccl(qibo_circ, datatype, pauli_string_pattern, n_sampl Returns: Expectation of quantum circuit due to pauli string. """ + import cuquantum.cutensornet as cutn from cupy.cuda import nccl from cuquantum import Network from mpi4py import MPI - import cuquantum.cutensornet as cutn root = 0 comm_mpi = MPI.COMM_WORLD @@ -368,9 +368,9 @@ def expectation_pauli_tn_MPI(qibo_circ, datatype, pauli_string_pattern, n_sample Returns: Expectation of quantum circuit due to pauli string. """ + import cuquantum.cutensornet as cutn from cuquantum import Network from mpi4py import MPI # this line initializes MPI - import cuquantum.cutensornet as cutn root = 0 comm = MPI.COMM_WORLD