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

Fix bug

Browse Source
This commit is contained in:
tankya2
2024-07-04 13:40:09 +08:00
parent e553bd28f7
commit 22a55281a5
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -14,8 +14,9 @@ class CuTensorNet(QibotnBackend, NumpyBackend): # pragma: no cover
def __init__(self, runcard): def __init__(self, runcard):
super().__init__() super().__init__()
import cuquantum
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:
self.MPI_enabled = runcard.get("MPI_enabled", False) self.MPI_enabled = runcard.get("MPI_enabled", False)
self.NCCL_enabled = runcard.get("NCCL_enabled", False) self.NCCL_enabled = runcard.get("NCCL_enabled", False)