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

Rename eval_tn_MPI_2 to eval_tn_MPI

Browse Source
This commit is contained in:
tankya2
2024-01-30 10:24:11 +08:00
parent 1035629745
commit 66aaf0e073
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/qibotn/cutn.py
View File

@@ -21,7 +21,7 @@ def eval_expectation(qibo_circ, datatype):
)
def eval_tn_MPI_2(qibo_circ, datatype, n_samples=8):
def eval_tn_MPI(qibo_circ, datatype, n_samples=8):
"""Convert qibo circuit to tensornet (TN) format and perform contraction using multi node and multi GPU through MPI.
The conversion is performed by QiboCircuitToEinsum(), after which it goes through 2 steps: pathfinder and execution.
The pathfinder looks at user defined number of samples (n_samples) iteratively to select the least costly contraction path. This is sped up with multi thread.