Remove import quantum

This commit is contained in:
tankya2
2024-10-30 11:06:23 +08:00
parent 616446cecc
commit 9890a45fb4

View File

@@ -14,7 +14,7 @@ class CuTensorNet(QibotnBackend, NumpyBackend): # pragma: no cover
def __init__(self, runcard):
super().__init__()
import cuquantum
from cuquantum import cudaDataType, ComputeType, __version__ # pylint: disable=import-error
from cuquantum import cutensornet as cutn # pylint: disable=import-error
if runcard is not None:
@@ -61,22 +61,21 @@ class CuTensorNet(QibotnBackend, NumpyBackend): # pragma: no cover
self.expectation_enabled = False
self.name = "qibotn"
self.cuquantum = cuquantum
self.cutn = cutn
self.platform = "cutensornet"
self.versions["cuquantum"] = self.cuquantum.__version__
self.versions["cuquantum"] = __version__
self.supports_multigpu = True
self.handle = self.cutn.create()
global CUDA_TYPES
CUDA_TYPES = {
"complex64": (
self.cuquantum.cudaDataType.CUDA_C_32F,
self.cuquantum.ComputeType.COMPUTE_32F,
cudaDataType.CUDA_C_32F,
ComputeType.COMPUTE_32F,
),
"complex128": (
self.cuquantum.cudaDataType.CUDA_C_64F,
self.cuquantum.ComputeType.COMPUTE_64F,
cudaDataType.CUDA_C_64F,
ComputeType.COMPUTE_64F,
),
}