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

Rename expectation function name

Browse Source
This commit is contained in:
tankya2
2024-01-31 10:02:34 +08:00
committed by yangliwei
parent 34acff9363
commit c43170ffa4
2 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -53,7 +53,7 @@ class CuTensorNet(NumpyBackend): # pragma: no cover
raise TypeError("Type can be either complex64 or complex128")
def execute_circuit(
self, circuit, test, MPI_enabled=False, MPS_enabled=False, NCCL_enabled=False, expectation_enabled=False, initial_state=None, nshots=None, return_array=False
self, circuit, MPI_enabled=False, MPS_enabled=False, NCCL_enabled=False, expectation_enabled=False, initial_state=None, nshots=None, return_array=False
): # pragma: no cover
"""Executes a quantum circuit.
@@ -68,7 +68,11 @@ class CuTensorNet(NumpyBackend): # pragma: no cover
"""
import qibotn.eval as eval
print("Test", test)
print("MPI_enabled", MPI_enabled)
print("MPS_enabled", MPS_enabled)
print("NCCL_enabled", NCCL_enabled)
print("expectation_enabled", expectation_enabled)
if (
MPI_enabled == False
and MPS_enabled == False
@@ -150,7 +154,7 @@ class CuTensorNet(NumpyBackend): # pragma: no cover
raise_error(NotImplementedError,
"QiboTN cannot support initial state.")
state, rank = eval.expectation_tn_MPI(
state, rank = eval.expectation_pauli_tn_MPI(
circuit, self.dtype, 32)
if rank > 0:
@@ -166,7 +170,7 @@ class CuTensorNet(NumpyBackend): # pragma: no cover
raise_error(NotImplementedError,
"QiboTN cannot support initial state.")
state, rank = eval.expectation_tn_nccl(
state, rank = eval.expectation_pauli_tn_nccl(
circuit, self.dtype, 32)
if rank > 0: