diff --git a/src/qibotn/backends/gpu.py b/src/qibotn/backends/gpu.py index 5f41344..6301bf7 100644 --- a/src/qibotn/backends/gpu.py +++ b/src/qibotn/backends/gpu.py @@ -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: diff --git a/src/qibotn/eval.py b/src/qibotn/eval.py index ddf2bfc..16ae046 100644 --- a/src/qibotn/eval.py +++ b/src/qibotn/eval.py @@ -15,7 +15,7 @@ def dense_vector_tn(qibo_circ, datatype): return contract(*myconvertor.state_vector_operands()) -def expectation_tn(qibo_circ, datatype): +def expectation_pauli_tn(qibo_circ, datatype): myconvertor = QiboCircuitToEinsum(qibo_circ, dtype=datatype) return contract( *myconvertor.expectation_operands(PauliStringGen(qibo_circ.nqubits)) @@ -204,7 +204,7 @@ def dense_vector_tn_nccl(qibo_circ, datatype, n_samples=8): return result, rank -def expectation_tn_nccl(qibo_circ, datatype, n_samples=8): +def expectation_pauli_tn_nccl(qibo_circ, datatype, n_samples=8): from mpi4py import MPI # this line initializes MPI import socket from cuquantum import Network @@ -291,7 +291,7 @@ def expectation_tn_nccl(qibo_circ, datatype, n_samples=8): return result, rank -def expectation_tn_MPI(qibo_circ, datatype, n_samples=8): +def expectation_pauli_tn_MPI(qibo_circ, datatype, n_samples=8): from mpi4py import MPI # this line initializes MPI import socket from cuquantum import Network