Update cuda_core/tests/test_device.py

Co-authored-by: Leo Fang <leof@nvidia.com>
NVIDIA · Nov 18, 2024 · 8daac0e · 8daac0e
1 parent c3e9927
commit 8daac0e
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/cuda_core/tests/test_device.py b/cuda_core/tests/test_device.py
@@ -7,13 +7,9 @@
 # is strictly prohibited.
 
 try:
-    from cuda.bindings import driver
+    from cuda.bindings import driver, runtime
 except ImportError:
     from cuda import cuda as driver
-
-try:
-    from cuda.bindings import runtime
-except ImportError:
     from cuda import cudart as runtime
 
 from cuda.core.experimental import Device