Import get_num_gpus in debug_utils

rapidsai · Jun 16, 2020 · 47e9977 · 47e9977
1 parent d93baf1
commit 47e9977
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/debug-tests/debug_utils.py b/debug-tests/debug_utils.py
@@ -6,6 +6,8 @@
 import cupy
 import rmm
 
+from .utils import get_num_gpus
+
 ITERATIONS = 100