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

correct missing mempool initialization

Browse Source
This commit is contained in:
tankya2
2024-07-04 16:10:43 +08:00
parent ef565eefc4
commit a5640a9d45
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/qibotn/eval.py
View File

@@ -320,6 +320,7 @@ def expectation_pauli_tn_MPI(qibo_circ, datatype, pauli_string_pattern, n_sample
""" """
from cuquantum import Network from cuquantum import Network
from mpi4py import MPI # this line initializes MPI from mpi4py import MPI # this line initializes MPI
import cuquantum.cutensornet as cutn
root = 0 root = 0
comm = MPI.COMM_WORLD comm = MPI.COMM_WORLD
@@ -329,6 +330,7 @@ def expectation_pauli_tn_MPI(qibo_circ, datatype, pauli_string_pattern, n_sample
# Assign the device for each process. # Assign the device for each process.
device_id = rank % getDeviceCount() device_id = rank % getDeviceCount()
cp.cuda.Device(device_id).use() cp.cuda.Device(device_id).use()
mempool = cp.get_default_memory_pool()
# Perform circuit conversion # Perform circuit conversion
if rank==0: if rank==0: