Skip to content

Commit fde1d3f

Browse files
committed
Fix version check
1 parent a55a771 commit fde1d3f

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

cuda_bindings/cuda/bindings/nvml.pyx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28457,7 +28457,7 @@ def __getattr__(name: str):
2845728457
# runtime and set the value accordingly.
2845828458

2845928459
if name == "NVLINK_MAX_LINKS":
28460-
if tuple(int(x) for x in system_get_nvml_version().split(".")) < (3, 13):
28460+
if tuple(int(x) for x in system_get_nvml_version().split(".")) < (3, 610):
2846128461
return 18
2846228462
else:
2846328463
return 36

cuda_bindings/tests/nvml/test_nvlink.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def test_nvlink_get_link_count(all_devices):
3737
def test_nvlink_max_links():
3838
nvml_version = tuple(int(x) for x in nvml.system_get_nvml_version().split("."))
3939

40-
if nvml_version < (13, 3):
40+
if nvml_version < (13, 610):
4141
assert nvml.NVLINK_MAX_LINKS == 18
4242
else:
4343
assert nvml.NVLINK_MAX_LINKS == 36

0 commit comments

Comments
 (0)