From 8d3e735956d9126ea3302f87cb887b1ad9e219d5 Mon Sep 17 00:00:00 2001 From: "Ralf W. Grosse-Kunstleve" Date: Fri, 4 Sep 2026 12:33:09 -0700 Subject: [PATCH] test(nvml): restore CUDA device-order assertions Remove the return accidentally retained when the Tegra skip check was converted to a loop. Supported systems now reach the CUDA/NVML device comparison, and the loop can check both Orin and Thor. --- cuda_bindings/tests/nvml/test_cuda.py | 1 - 1 file changed, 1 deletion(-) diff --git a/cuda_bindings/tests/nvml/test_cuda.py b/cuda_bindings/tests/nvml/test_cuda.py index c9997af45fb..ed260a2fbc6 100644 --- a/cuda_bindings/tests/nvml/test_cuda.py +++ b/cuda_bindings/tests/nvml/test_cuda.py @@ -61,7 +61,6 @@ def test_cuda_device_order(): for kind in ("Orin", "Thor"): if any(kind in device["name"] for device in nvml_devices): pytest.skip(f"Skipping test on {kind}, which has non-standard device naming") - return if "CUDA_VISIBLE_DEVICES" not in os.environ: # If that environment variable isn't set, the device lists should match exactly