diff --git a/src/qibotn/backends/cutensornet.py b/src/qibotn/backends/cutensornet.py index 7696919..7dd1091 100644 --- a/src/qibotn/backends/cutensornet.py +++ b/src/qibotn/backends/cutensornet.py @@ -1,10 +1,10 @@ import numpy as np +from qibo import hamiltonians from qibo.backends import NumpyBackend from qibo.config import raise_error -from qibotn.result import TensorNetworkResult from qibotn.backends.abstract import QibotnBackend -from qibo import hamiltonians +from qibotn.result import TensorNetworkResult CUDA_TYPES = {} diff --git a/src/qibotn/eval.py b/src/qibotn/eval.py index 0aae2ca..bb5b104 100644 --- a/src/qibotn/eval.py +++ b/src/qibotn/eval.py @@ -4,18 +4,13 @@ from cupy.cuda import nccl from cupy.cuda.runtime import getDeviceCount from cuquantum import Network, contract from mpi4py import MPI +from qibo import hamiltonians +from qibo.symbols import I, X, Y, Z from qibotn.circuit_convertor import QiboCircuitToEinsum from qibotn.circuit_to_mps import QiboCircuitToMPS 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 -from qibo import hamiltonians -from qibo.symbols import X, Y, Z, I - def check_observable(observable, circuit_nqubit): """Checks the type of observable and returns the appropriate Hamiltonian.""" diff --git a/tests/test_cuquantum_cutensor_backend.py b/tests/test_cuquantum_cutensor_backend.py index dffea58..5130dd5 100644 --- a/tests/test_cuquantum_cutensor_backend.py +++ b/tests/test_cuquantum_cutensor_backend.py @@ -1,13 +1,13 @@ +import math from timeit import default_timer as timer import cupy as cp import numpy as np import pytest import qibo +from qibo import Circuit, construct_backend, gates, hamiltonians from qibo.models import QFT -from qibo import Circuit, gates, hamiltonians, construct_backend from qibo.symbols import X, Z -import math def qibo_qft(nqubits, swaps):