Remove duplicate functions from debug_utils

rapidsai · Jun 15, 2020 · d93baf1 · d93baf1
1 parent b62d359
commit d93baf1
Showing 1 changed file with 0 additions and 13 deletions.
diff --git a/debug-tests/debug_utils.py b/debug-tests/debug_utils.py
@@ -9,10 +9,6 @@
 ITERATIONS = 100
 
 
-def cuda_array(size):
-    return rmm.DeviceBuffer(size=size)
-
-
 def set_rmm():
     rmm.reinitialize(
         pool_allocator=True, managed_memory=False, initial_pool_size=parse_bytes("6GB")
@@ -41,15 +37,6 @@ def parse_args(args=None):
     return parser.parse_args()
 
 
-def get_num_gpus():
-    import pynvml
-
-    pynvml.nvmlInit()
-    ngpus = pynvml.nvmlDeviceGetCount()
-    pynvml.nvmlShutdown()
-    return ngpus
-
-
 def get_cuda_devices():
     if "CUDA_VISIBLE_DEVICES" in os.environ:
         return os.environ["CUDA_VISIBLE_DEVICES"].split(",")