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

Remove import quantum

Browse Source
This commit is contained in:
tankya2
2024-10-30 11:06:23 +08:00
parent 616446cecc
commit 9890a45fb4
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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