This commit is contained in:
tankya2
2024-07-04 13:40:09 +08:00
parent e553bd28f7
commit 22a55281a5

View File

@@ -14,8 +14,9 @@ class CuTensorNet(QibotnBackend, NumpyBackend): # pragma: no cover
def __init__(self, runcard):
super().__init__()
import cuquantum
from cuquantum import cutensornet as cutn # pylint: disable=import-error
if runcard is not None:
self.MPI_enabled = runcard.get("MPI_enabled", False)
self.NCCL_enabled = runcard.get("NCCL_enabled", False)