test: skip system NVML tests when the symbol is absent in older bindings

Author: rluo8

cuda_core/tests/system/conftest.py

@@ -3,6 +3,8 @@
 # SPDX-License-Identifier: Apache-2.0
 
 
+from contextlib import contextmanager
+
 import pytest
 
 from cuda.core import system
@@ -22,7 +24,24 @@
 )
 
 
+@contextmanager
 def unsupported_before(device, expected_device_arch):
+    from cuda.bindings import nvml
+    from cuda.bindings._internal.utils import FunctionNotFoundError
     from cuda.bindings._test_helpers.arch_check import unsupported_before as nvml_unsupported_before
 
-    return nvml_unsupported_before(device._handle, expected_device_arch)
+    try:
+        with nvml_unsupported_before(device._handle, expected_device_arch):
+            yield
+    except FunctionNotFoundError:
+        # Backstop older cuda.bindings versions that don't skip when an NVML symbol is
+        # absent from the loaded NVML library. Only skip on the "possibly unsupported"
+        # branch, matching arch_check.py.
+        possibly_unsupported = (
+            expected_device_arch is None
+            or expected_device_arch == "HAS_INFOROM"
+            or nvml.device_get_architecture(device._handle) == nvml.DeviceArch.UNKNOWN
+        )
+        if possibly_unsupported:
+            pytest.skip("Requested NVML symbol is missing from the loaded driver/library")
+        raise