Fix nvbug6084457: Make NVLINK_MAX_LINKS version-dependent

Author: mdboom

cuda_bindings/cuda/bindings/nvml.pyx

@@ -1593,7 +1593,11 @@ class FieldId(_FastEnum):
 
     MAX = 289
 
-NVLINK_MAX_LINKS = 18
+
+if tuple(int(x) for x in system_get_nvml_version().split(".")) < (3, 13):
+    NVLINK_MAX_LINKS = 18
+else:
+    NVLINK_MAX_LINKS = 36
 
 
 class RUSD(_FastEnum):

cuda_core/tests/system/test_system_device.py

@@ -762,10 +762,18 @@ def test_compute_running_processes():
 
 
 def test_nvlink():
+    from cuda.bindings._version import version_tuple as cuda_bindings_version
+
+    nvml_version = system.get_nvml_version()
+
     for device in system.Device.get_all_devices():
         max_links = _device.NvlinkInfo.max_links
         assert isinstance(max_links, int)
         assert max_links > 0
+        if nvml_version >= (3, 13) and cuda_bindings_version > (13, 3, 1):
+            assert max_links == 36
+        else:
+            assert max_links == 18
 
         for link in range(max_links):
             with unsupported_before(device, None):