[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
committed by
tankya2
parent
ac396a35db
commit
4b44b10d6b
@@ -1,16 +1,14 @@
|
|||||||
import cupy as cp
|
import cupy as cp
|
||||||
|
import cuquantum.cutensornet as cutn
|
||||||
|
from cupy.cuda import nccl
|
||||||
from cupy.cuda.runtime import getDeviceCount
|
from cupy.cuda.runtime import getDeviceCount
|
||||||
from cuquantum import contract
|
from cuquantum import Network, contract
|
||||||
|
from mpi4py import MPI
|
||||||
|
|
||||||
from qibotn.circuit_convertor import QiboCircuitToEinsum
|
from qibotn.circuit_convertor import QiboCircuitToEinsum
|
||||||
from qibotn.circuit_to_mps import QiboCircuitToMPS
|
from qibotn.circuit_to_mps import QiboCircuitToMPS
|
||||||
from qibotn.mps_contraction_helper import MPSContractionHelper
|
from qibotn.mps_contraction_helper import MPSContractionHelper
|
||||||
|
|
||||||
import cuquantum.cutensornet as cutn
|
|
||||||
from cuquantum import Network
|
|
||||||
from mpi4py import MPI
|
|
||||||
from cupy.cuda import nccl
|
|
||||||
|
|
||||||
|
|
||||||
def initialize_mpi():
|
def initialize_mpi():
|
||||||
"""Initialize MPI communication and device selection."""
|
"""Initialize MPI communication and device selection."""
|
||||||
|
|||||||
Reference in New Issue
Block a user