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

Update to allow user to set MPS parameters and to set Pauli string pattern

Browse Source
This commit is contained in:
tankya2
2024-01-31 16:47:10 +08:00
committed by yangliwei
parent 784b1a70ef
commit 05f8523649
2 changed files with 44 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/qibotn/eval.py
View File

@@ -18,7 +18,9 @@ def dense_vector_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, pauli_string))
*myconvertor.expectation_operands(
PauliStringGen(qibo_circ.nqubits, pauli_string)
)
)
@@ -232,7 +234,9 @@ def expectation_pauli_tn_nccl(qibo_circ, datatype, pauli_string, 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, pauli_string))
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)
@@ -311,7 +315,9 @@ def expectation_pauli_tn_MPI(qibo_circ, datatype, pauli_string, 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, pauli_string))
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)
@@ -384,7 +390,7 @@ def PauliStringGen(nqubits, pauli_string):
return "Invalid input. N should be a positive integer."
characters = pauli_string
#characters = "XXXZ"
# characters = "XXXZ"
result = ""