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

Remove redundant mem pool

Browse Source
This commit is contained in:
tankya2
2024-12-18 14:41:41 +08:00
parent 9381ae6b88
commit 84ec5cfdfe
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/qibotn/eval.py
View File

@@ -73,7 +73,6 @@ def dense_vector_tn_MPI(qibo_circ, datatype, n_samples=8):
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:
@@ -160,7 +159,6 @@ def dense_vector_tn_nccl(qibo_circ, datatype, n_samples=8):
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()
# Set up the NCCL communicator. # Set up the NCCL communicator.
nccl_id = nccl.get_unique_id() if rank == root else None nccl_id = nccl.get_unique_id() if rank == root else None
@@ -263,7 +261,6 @@ def expectation_pauli_tn_nccl(qibo_circ, datatype, pauli_string_pattern, n_sampl
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()
# Set up the NCCL communicator. # Set up the NCCL communicator.
nccl_id = nccl.get_unique_id() if rank == root else None nccl_id = nccl.get_unique_id() if rank == root else None
@@ -368,7 +365,6 @@ 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: