Skip NVML tests if the hardware doesn't support NVML (#1585) (#1749)

* Skip NVML tests if the hardware doesn't support NVML

* Also skip converting between Device types

* Fix test_to_system_device test

* Update cuda_bindings/cuda/bindings/_test_helpers/arch_check.py



* Fix test

* Simplify check

* Fix import when cuda_bindings is incompatible

* Fix logic

* Move over a single test from #1583

* Fix check

---------

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: mdboom

cuda_bindings/cuda/bindings/_test_helpers/arch_check.py

@@ -3,12 +3,31 @@
 
 
 from contextlib import contextmanager
+from functools import cache
 
 import pytest
 
 from cuda.bindings import nvml
 
 
+@cache
+def hardware_supports_nvml():
+    """
+    Tries to call the simplest NVML API possible to see if just the basics
+    works.  If not we are probably on one of the platforms where NVML is not
+    supported at all (e.g. Jetson Orin).
+    """
+    nvml.init_v2()
+    try:
+        nvml.system_get_driver_branch()
+    except (nvml.NotSupportedError, nvml.UnknownError):
+        return False
+    else:
+        return True
+    finally:
+        nvml.shutdown()
+
+
 @contextmanager
 def unsupported_before(device: int, expected_device_arch: nvml.DeviceArch | str | None):
     device_arch = nvml.device_get_architecture(device)

cuda_bindings/tests/nvml/__init__.py

@@ -1,2 +1,10 @@
-# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
+
+
+import pytest
+
+from cuda.bindings._test_helpers.arch_check import hardware_supports_nvml
+
+if not hardware_supports_nvml():
+    pytest.skip("NVML not supported on this platform", allow_module_level=True)