[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot]
2024-10-04 07:26:39 +00:00
committed by tankya2
parent c3a42f8760
commit 616446cecc
2 changed files with 5 additions and 5 deletions

View File

@@ -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)

View File

@@ -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