Support systems with old cuda.bindings

Author: mdboom

cuda_core/cuda/core/_device.pyx

@@ -1047,6 +1047,13 @@ class Device:
         cuda.core.system.Device
             The corresponding system-level device instance used for NVML access.
         """
+        from cuda.core.system._system import CUDA_BINDINGS_NVML_IS_COMPATIBLE
+
+        if not CUDA_BINDINGS_NVML_IS_COMPATIBLE:
+            raise RuntimeError(
+                "cuda.core.system.Device requires cuda_bindings 13.1.2+ or 12.9.6+"
+            )
+
         from cuda.core.system import Device as SystemDevice
         return SystemDevice(uuid=self.uuid)
 

cuda_core/tests/test_device.py

@@ -26,9 +26,17 @@ def cuda_version():
 
 
 def test_to_system_device(deinit_cuda):
-    from cuda.core.system import Device as SystemDevice
+    from cuda.core.system import _system
 
     device = Device()
+
+    if not _system.CUDA_BINDINGS_NVML_IS_COMPATIBLE:
+        with pytest.raises(RuntimeError):
+            device.to_system_device()
+        pytest.skip("NVML support requires cuda.bindings version 12.9.6+ or 13.1.2+")
+
+    from cuda.core.system import Device as SystemDevice
+
     system_device = device.to_system_device()
     assert isinstance(system_device, SystemDevice)
     assert system_device.uuid[4:] == device.uuid