From 616446cecc6a4050c5ac0cc464e85dcd971ae70a Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 4 Oct 2024 07:26:39 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- src/qibotn/backends/cutensornet.py | 2 +- src/qibotn/eval.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) 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