Skip to content

Commit c789bf6

Browse files
committed
Run some assertions only if _BINDING_VERSION >= (12, 0)
1 parent 8bdec20 commit c789bf6

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

cuda_core/tests/test_wip.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
11
from cuda.bindings import driver, nvrtc, runtime
22
from cuda.core.experimental import _utils
33

4-
err, _DRIVER_VERSION = driver.cuDriverGetVersion()
5-
assert err == driver.CUresult.CUDA_SUCCESS
4+
_BINDING_VERSION = _utils.get_binding_version()
65

76

87
def test_driver_error_info():
@@ -12,7 +11,7 @@ def test_driver_error_info():
1211
try:
1312
error = driver.CUresult(code)
1413
except ValueError:
15-
if _DRIVER_VERSION >= 12000:
14+
if _BINDING_VERSION >= (12, 0):
1615
assert code not in expl_dict
1716
else:
1817
assert code in expl_dict
@@ -25,8 +24,9 @@ def test_driver_error_info():
2524
print(desc)
2625
print(expl)
2726
print()
28-
stray_expl_codes = sorted(set(expl_dict.keys()) - valid_codes)
29-
assert not stray_expl_codes
27+
if _BINDING_VERSION >= (12, 0):
28+
extra_expl_codes = sorted(set(expl_dict.keys()) - valid_codes)
29+
assert not extra_expl_codes
3030
missing_expl_codes = sorted(valid_codes - set(expl_dict.keys()))
3131
assert not missing_expl_codes
3232

@@ -38,7 +38,7 @@ def test_runtime_error_info():
3838
try:
3939
error = runtime.cudaError_t(code)
4040
except ValueError:
41-
if _DRIVER_VERSION >= 12000:
41+
if _BINDING_VERSION >= (12, 0):
4242
assert code not in expl_dict
4343
else:
4444
assert code in expl_dict
@@ -51,8 +51,9 @@ def test_runtime_error_info():
5151
print(desc)
5252
print(expl)
5353
print()
54-
stray_expl_codes = sorted(set(expl_dict.keys()) - valid_codes)
55-
assert not stray_expl_codes
54+
if _BINDING_VERSION >= (12, 0):
55+
extra_expl_codes = sorted(set(expl_dict.keys()) - valid_codes)
56+
assert not extra_expl_codes
5657
missing_expl_codes = sorted(valid_codes - set(expl_dict.keys()))
5758
assert not missing_expl_codes
5859

0 commit comments

Comments
 (0)