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

Allow user to specify Pauli string pattern for expecation calculation [skip CI]

Browse Source
This commit is contained in:
tankya2
2024-01-31 14:56:21 +08:00
committed by yangliwei
parent 928f99e336
commit 784b1a70ef
2 changed files with 38 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/qibotn/eval.py
View File

@@ -15,10 +15,10 @@ def dense_vector_tn(qibo_circ, datatype):
return contract(*myconvertor.state_vector_operands())
def expectation_pauli_tn(qibo_circ, datatype):
def expectation_pauli_tn(qibo_circ, datatype, pauli_string):
myconvertor = QiboCircuitToEinsum(qibo_circ, dtype=datatype)
return contract(
*myconvertor.expectation_operands(PauliStringGen(qibo_circ.nqubits))
*myconvertor.expectation_operands(PauliStringGen(qibo_circ.nqubits, pauli_string))
)
@@ -204,7 +204,7 @@ def dense_vector_tn_nccl(qibo_circ, datatype, n_samples=8):
return result, rank
def expectation_pauli_tn_nccl(qibo_circ, datatype, n_samples=8):
def expectation_pauli_tn_nccl(qibo_circ, datatype, pauli_string, n_samples=8):
from mpi4py import MPI # this line initializes MPI
import socket
from cuquantum import Network
@@ -232,7 +232,7 @@ def expectation_pauli_tn_nccl(qibo_circ, datatype, n_samples=8):
myconvertor = QiboCircuitToEinsum(qibo_circ, dtype=datatype)
# mem_avail = cp.cuda.Device().mem_info[0]
# print("Mem avail: aft convetor",mem_avail, "rank =",rank)
operands = myconvertor.expectation_operands(PauliStringGen(qibo_circ.nqubits))
operands = myconvertor.expectation_operands(PauliStringGen(qibo_circ.nqubits, pauli_string))
# mem_avail = cp.cuda.Device().mem_info[0]
# print("Mem avail: aft operand interleave",mem_avail, "rank =",rank)
@@ -291,7 +291,7 @@ def expectation_pauli_tn_nccl(qibo_circ, datatype, n_samples=8):
return result, rank
def expectation_pauli_tn_MPI(qibo_circ, datatype, n_samples=8):
def expectation_pauli_tn_MPI(qibo_circ, datatype, pauli_string, n_samples=8):
from mpi4py import MPI # this line initializes MPI
import socket
from cuquantum import Network
@@ -311,7 +311,7 @@ def expectation_pauli_tn_MPI(qibo_circ, datatype, n_samples=8):
myconvertor = QiboCircuitToEinsum(qibo_circ, dtype=datatype)
# mem_avail = cp.cuda.Device().mem_info[0]
# print("Mem avail: aft convetor",mem_avail, "rank =",rank)
operands = myconvertor.expectation_operands(PauliStringGen(qibo_circ.nqubits))
operands = myconvertor.expectation_operands(PauliStringGen(qibo_circ.nqubits, pauli_string))
# mem_avail = cp.cuda.Device().mem_info[0]
# print("Mem avail: aft operand interleave",mem_avail, "rank =",rank)
@@ -379,17 +379,17 @@ def dense_vector_mps(qibo_circ, gate_algo, datatype):
)
def PauliStringGen(nqubits):
def PauliStringGen(nqubits, pauli_string):
if nqubits <= 0:
return "Invalid input. N should be a positive integer."
# characters = 'IXYZ'
characters = "XXXZ"
characters = pauli_string
#characters = "XXXZ"
result = ""
for i in range(nqubits):
char_to_add = characters[i % len(characters)]
result += char_to_add
print("pauli string", result)
return result