From 72031e34aa9c557b8e966ee6dd4d5548e06dc18d Mon Sep 17 00:00:00 2001 From: ksimpson Date: Tue, 21 Jan 2025 12:00:55 -0800 Subject: [PATCH] update test --- cuda_core/tests/test_module.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/cuda_core/tests/test_module.py b/cuda_core/tests/test_module.py index 7dfd5ba2..aaad63b6 100644 --- a/cuda_core/tests/test_module.py +++ b/cuda_core/tests/test_module.py @@ -11,9 +11,8 @@ from conftest import can_load_generated_ptx try: - from cuda.bindings import driver, runtime + from cuda.bindings import runtime except ImportError: - from cuda import cuda as driver from cuda import cudart as runtime from cuda.core.experimental import Program, ProgramOptions @@ -22,7 +21,9 @@ @pytest.fixture(scope="module") def cuda_version(): - version = handle_return(driver.cuDriverGetVersion()) + # WAR this is a workaround for the fact that checking the runtime version using the cuGetDriverVersion + # doesnt actually return the driver version buyt rather the latest cuda whcih is supported by the installed drive3r. + version = handle_return(runtime.cudaRuntimeGetVersion()) major_version = version // 1000 minor_version = (version % 1000) // 10