From 22a55281a5959d1dcf79c181a4f8b241d7f73344 Mon Sep 17 00:00:00 2001 From: tankya2 Date: Thu, 4 Jul 2024 13:40:09 +0800 Subject: [PATCH] Fix bug --- src/qibotn/backends/cutensornet.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/qibotn/backends/cutensornet.py b/src/qibotn/backends/cutensornet.py index 553fc51..4f1c409 100644 --- a/src/qibotn/backends/cutensornet.py +++ b/src/qibotn/backends/cutensornet.py @@ -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)