More precise version checks

Author: mdboom

cuda_core/cuda/core/experimental/system/__init__.py

@@ -21,7 +21,11 @@
 # We need both the existence of cuda.bindings._nvml and a sufficient version
 # with the APIs implemented as we need them.
 
-if tuple(int(x) for x in cuda.bindings.__version__.split(".")) >= (13, 1, 1):
+_BINDINGS_VERSION = tuple(int(x) for x in cuda.bindings.__version__.split("."))
+
+if (_BINDINGS_VERSION[0] == 13 and _BINDINGS_VERSION[1:3] >= (1, 1)) or (
+    _BINDINGS_VERSION[0] == 12 and _BINDINGS_VERSION[1:3] >= (9, 5)
+):
     from cuda.bindings import _nvml
 
     from ._nvml_context import initialize

cuda_core/tests/system/conftest.py

@@ -9,7 +9,9 @@
 CUDA_BINDINGS_VERSION = tuple(int(x) for x in cuda.bindings.__version__.split("."))
 
 
-NVML_SUPPORTED = CUDA_BINDINGS_VERSION >= (13, 1, 1)
+NVML_SUPPORTED = (CUDA_BINDINGS_VERSION[0] == 13 and CUDA_BINDINGS_VERSION[1:3] >= (1, 1)) or (
+    CUDA_BINDINGS_VERSION[0] == 12 and CUDA_BINDINGS_VERSION[1:3] >= (9, 5)
+)
 
 
 skip_if_nvml_unsupported = pytest.mark.skipif(