test: add gpu sanity test for ci job (#49)

examples/tests/test_gpu_sanity.py

+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import subprocess
+
+import pytest
+
+pytestmark = pytest.mark.gpu
+
+
+def test_detect_gpu():
+    try:
+        result = subprocess.run(
+            ["nvidia-smi"], capture_output=True, text=True, check=True
+        )
+        print("\nAvailable GPUs:")
+        print(result.stdout)
+        assert "NVIDIA" in result.stdout, "No NVIDIA GPUs found in nvidia-smi output"
+    except subprocess.CalledProcessError as e:
+        pytest.fail(f"nvidia-smi command failed with error: {e}")
+    except FileNotFoundError:
+        pytest.fail(
+            "nvidia-smi command not found. Ensure NVIDIA drivers are properly installed."
+        )

pyproject.toml

@@ -105,7 +105,8 @@ asyncio_mode = "auto"
 markers = [
   "pre_merge: marks tests to run before merging",
   "nightly: marks tests to run nightly",
-  "weekly: marks tests to run weekly"
+  "weekly: marks tests to run weekly",
+  "gpu: marks tests to run on GPU"
 ]
 
 # Linting/formatting