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

Remove print [skip CI]

Browse Source
This commit is contained in:
tankya2
2024-01-31 10:56:47 +08:00
parent 168018593c
commit e7776f55bf
1 changed files with 1 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/qibotn/backends/gpu.py
View File

@@ -14,8 +14,6 @@ class CuTensorNet(NumpyBackend): # pragma: no cover
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:
print("inside runcard")
# Parse the runcard or use its values to set flags
self.MPI_enabled = runcard.get("MPI_enabled", False) self.MPI_enabled = runcard.get("MPI_enabled", False)
self.MPS_enabled = runcard.get("MPS_enabled", False) self.MPS_enabled = runcard.get("MPS_enabled", False)
self.NCCL_enabled = runcard.get("NCCL_enabled", False) self.NCCL_enabled = runcard.get("NCCL_enabled", False)
@@ -82,11 +80,6 @@ class CuTensorNet(NumpyBackend): # pragma: no cover
import qibotn.eval as eval import qibotn.eval as eval
print("MPI_enabled", self.MPI_enabled)
print("MPS_enabled", self.MPS_enabled)
print("NCCL_enabled", self.NCCL_enabled)
print("expectation_enabled", self.expectation_enabled)
if ( if (
self.MPI_enabled == False self.MPI_enabled == False
and self.MPS_enabled == False and self.MPS_enabled == False
@@ -181,7 +174,7 @@ class CuTensorNet(NumpyBackend): # pragma: no cover
if rank > 0: if rank > 0:
state = np.array(0) state = np.array(0)
else: else:
raise_error(NotImplementedError, "Backend not supported.") raise_error(NotImplementedError, "Compute type not supported.")
if return_array: if return_array:
return state.flatten() return state.flatten()