Skip to content

Commit

Permalink
update test
Browse files Browse the repository at this point in the history
  • Loading branch information
ksimpson-work committed Jan 21, 2025
1 parent caebb92 commit 72031e3
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions cuda_core/tests/test_module.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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
Expand Down

0 comments on commit 72031e3

Please sign in to comment.