From 8daac0ecee32db05dc635c85e28946e8550c6e9c Mon Sep 17 00:00:00 2001 From: Keenan Simpson Date: Mon, 18 Nov 2024 11:40:30 -0800 Subject: [PATCH] Update cuda_core/tests/test_device.py Co-authored-by: Leo Fang --- cuda_core/tests/test_device.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/cuda_core/tests/test_device.py b/cuda_core/tests/test_device.py index 9aadf0db..c809bfb3 100644 --- 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