From 84ec5cfdfed5436a7efbc306cf88f1d2713a437c Mon Sep 17 00:00:00 2001 From: tankya2 Date: Wed, 18 Dec 2024 14:41:41 +0800 Subject: [PATCH] Remove redundant mem pool --- src/qibotn/eval.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/src/qibotn/eval.py b/src/qibotn/eval.py index d9d93c4..ebef103 100644 --- a/src/qibotn/eval.py +++ b/src/qibotn/eval.py @@ -73,7 +73,6 @@ def dense_vector_tn_MPI(qibo_circ, datatype, n_samples=8): device_id = rank % getDeviceCount() cp.cuda.Device(device_id).use() - mempool = cp.get_default_memory_pool() # Perform circuit conversion if rank == 0: @@ -160,7 +159,6 @@ def dense_vector_tn_nccl(qibo_circ, datatype, n_samples=8): device_id = rank % getDeviceCount() cp.cuda.Device(device_id).use() - mempool = cp.get_default_memory_pool() # Set up the NCCL communicator. 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() cp.cuda.Device(device_id).use() - mempool = cp.get_default_memory_pool() # Set up the NCCL communicator. 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. device_id = rank % getDeviceCount() cp.cuda.Device(device_id).use() - mempool = cp.get_default_memory_pool() # Perform circuit conversion if rank == 0: