Skip to content

Commit bd98f21

Browse files
committed
Fix test
1 parent cfb3a03 commit bd98f21

1 file changed

Lines changed: 8 additions & 3 deletions

File tree

cuda_bindings/tests/test_cudart.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99

1010
import cuda.bindings.driver as cuda
1111
import cuda.bindings.runtime as cudart
12+
from cuda import pathfinder
1213
from cuda.bindings import runtime
1314

1415

@@ -1403,6 +1404,10 @@ def test_struct_pointer_comparison(target):
14031404

14041405

14051406
def test_getLocalRuntimeVersion():
1406-
err, version = cudart.getLocalRuntimeVersion()
1407-
assertSuccess(err)
1408-
assert version >= 12000 # CUDA 12.0
1407+
try:
1408+
err, version = cudart.getLocalRuntimeVersion()
1409+
except pathfinder.DynamicLibraryNotFoundError:
1410+
pass
1411+
else:
1412+
assertSuccess(err)
1413+
assert version >= 12000 # CUDA 12.0

0 commit comments

Comments
 (0)