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:

6
src/qibotn/eval.py
View File

@@ -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