jaunatisblue/final-qibotn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

2
src/qibotn/backends/cutensornet.py
View File

@@ -16,7 +16,7 @@ class CuTensorNet(QibotnBackend, NumpyBackend): # pragma: no cover
super().__init__() super().__init__()
import cuquantum import cuquantum
from cuquantum import cutensornet as cutn # pylint: disable=import-error from cuquantum import cutensornet as cutn # pylint: disable=import-error
if runcard is not None: if runcard is not None:
self.MPI_enabled = runcard.get("MPI_enabled", False) self.MPI_enabled = runcard.get("MPI_enabled", False)
self.NCCL_enabled = runcard.get("NCCL_enabled", False) self.NCCL_enabled = runcard.get("NCCL_enabled", False)

8
src/qibotn/eval.py
View File

@@ -62,9 +62,9 @@ def dense_vector_tn_MPI(qibo_circ, datatype, n_samples=8):
Dense vector of quantum circuit. Dense vector of quantum circuit.
""" """
import cuquantum.cutensornet as cutn
from cuquantum import Network from cuquantum import Network
from mpi4py import MPI from mpi4py import MPI
import cuquantum.cutensornet as cutn
root = 0 root = 0
comm = MPI.COMM_WORLD comm = MPI.COMM_WORLD
@@ -150,10 +150,10 @@ def dense_vector_tn_nccl(qibo_circ, datatype, n_samples=8):
Returns: Returns:
Dense vector of quantum circuit. Dense vector of quantum circuit.
""" """
import cuquantum.cutensornet as cutn
from cupy.cuda import nccl from cupy.cuda import nccl
from cuquantum import Network from cuquantum import Network
from mpi4py import MPI from mpi4py import MPI
import cuquantum.cutensornet as cutn
root = 0 root = 0
comm_mpi = MPI.COMM_WORLD comm_mpi = MPI.COMM_WORLD
@@ -259,10 +259,10 @@ def expectation_pauli_tn_nccl(qibo_circ, datatype, pauli_string_pattern, n_sampl
Returns: Returns:
Expectation of quantum circuit due to pauli string. Expectation of quantum circuit due to pauli string.
""" """
import cuquantum.cutensornet as cutn
from cupy.cuda import nccl from cupy.cuda import nccl
from cuquantum import Network from cuquantum import Network
from mpi4py import MPI from mpi4py import MPI
import cuquantum.cutensornet as cutn
root = 0 root = 0
comm_mpi = MPI.COMM_WORLD comm_mpi = MPI.COMM_WORLD
@@ -368,9 +368,9 @@ def expectation_pauli_tn_MPI(qibo_circ, datatype, pauli_string_pattern, n_sample
Returns: Returns:
Expectation of quantum circuit due to pauli string. Expectation of quantum circuit due to pauli string.
""" """
import cuquantum.cutensornet as cutn
from cuquantum import Network from cuquantum import Network
from mpi4py import MPI # this line initializes MPI from mpi4py import MPI # this line initializes MPI
import cuquantum.cutensornet as cutn
root = 0 root = 0
comm = MPI.COMM_WORLD comm = MPI.COMM_WORLD