We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a55a771 commit fde1d3fCopy full SHA for fde1d3f
2 files changed
cuda_bindings/cuda/bindings/nvml.pyx
@@ -28457,7 +28457,7 @@ def __getattr__(name: str):
28457
# runtime and set the value accordingly.
28458
28459
if name == "NVLINK_MAX_LINKS":
28460
- if tuple(int(x) for x in system_get_nvml_version().split(".")) < (3, 13):
+ if tuple(int(x) for x in system_get_nvml_version().split(".")) < (3, 610):
28461
return 18
28462
else:
28463
return 36
cuda_bindings/tests/nvml/test_nvlink.py
@@ -37,7 +37,7 @@ def test_nvlink_get_link_count(all_devices):
37
def test_nvlink_max_links():
38
nvml_version = tuple(int(x) for x in nvml.system_get_nvml_version().split("."))
39
40
- if nvml_version < (13, 3):
+ if nvml_version < (13, 610):
41
assert nvml.NVLINK_MAX_LINKS == 18
42
43
assert nvml.NVLINK_MAX_LINKS == 36
0 commit comments