[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
import cupy as cp
|
||||
from cuquantum.cutensornet.experimental import contract_decompose
|
||||
from cuquantum import contract
|
||||
from cuquantum.cutensornet.experimental import contract_decompose
|
||||
|
||||
|
||||
def initial(num_qubits, dtype):
|
||||
|
||||
@@ -94,8 +94,7 @@ class QiboCircuitToEinsum:
|
||||
required_shape = self.op_shape_from_qubits(len(gate_qubits))
|
||||
self.gate_tensors.append(
|
||||
(
|
||||
cp.asarray(gate.matrix(), dtype=self.dtype).reshape(
|
||||
required_shape),
|
||||
cp.asarray(gate.matrix(), dtype=self.dtype).reshape(required_shape),
|
||||
gate_qubits,
|
||||
)
|
||||
)
|
||||
|
||||
@@ -1,9 +1,9 @@
|
||||
import cupy as cp
|
||||
import numpy as np
|
||||
|
||||
from cuquantum import cutensornet as cutn
|
||||
|
||||
from qibotn.MPSUtils import apply_gate, initial
|
||||
from qibotn.QiboCircuitConvertor import QiboCircuitToEinsum
|
||||
from qibotn.MPSUtils import initial, apply_gate
|
||||
|
||||
|
||||
class QiboCircuitToMPS:
|
||||
|
||||
@@ -1,12 +1,13 @@
|
||||
from qibotn.QiboCircuitConvertor import QiboCircuitToEinsum
|
||||
import multiprocessing
|
||||
|
||||
import cupy as cp
|
||||
from cupy.cuda.runtime import getDeviceCount
|
||||
from cuquantum import contract
|
||||
from cuquantum import cutensornet as cutn
|
||||
import multiprocessing
|
||||
from cupy.cuda.runtime import getDeviceCount
|
||||
import cupy as cp
|
||||
|
||||
from qibotn.QiboCircuitToMPS import QiboCircuitToMPS
|
||||
from qibotn.mps_contraction_helper import MPSContractionHelper
|
||||
from qibotn.QiboCircuitConvertor import QiboCircuitToEinsum
|
||||
from qibotn.QiboCircuitToMPS import QiboCircuitToMPS
|
||||
|
||||
|
||||
def eval(qibo_circ, datatype):
|
||||
@@ -49,7 +50,7 @@ def eval_tn_MPI(qibo_circ, datatype, n_samples=8):
|
||||
|
||||
return result, rank
|
||||
|
||||
|
||||
|
||||
def eval_mps(qibo_circ, gate_algo, datatype):
|
||||
myconvertor = QiboCircuitToMPS(qibo_circ, gate_algo, dtype=datatype)
|
||||
mps_helper = MPSContractionHelper(myconvertor.num_qubits)
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
from cuquantum import contract, contract_path, CircuitToEinsum, tensor
|
||||
from cuquantum import CircuitToEinsum, contract, contract_path, tensor
|
||||
|
||||
|
||||
class MPSContractionHelper:
|
||||
|
||||
@@ -3,8 +3,14 @@ import quimb.tensor as qtn
|
||||
from qibo.models import Circuit as QiboCircuit
|
||||
|
||||
|
||||
def from_qibo(circuit: QiboCircuit, is_mps: False, psi0=None, method='svd',
|
||||
cutoff=1e-6, cutoff_mode='abs'):
|
||||
def from_qibo(
|
||||
circuit: QiboCircuit,
|
||||
is_mps: False,
|
||||
psi0=None,
|
||||
method="svd",
|
||||
cutoff=1e-6,
|
||||
cutoff_mode="abs",
|
||||
):
|
||||
nqubits = circuit.nqubits
|
||||
gate_opt = {}
|
||||
if is_mps:
|
||||
|
||||
Reference in New Issue
Block a user