Add PEP723 metadata to cuda_core examples

Author: mdboom

.github/workflows/test-wheel-linux.yml

@@ -96,6 +96,9 @@ jobs:
         uses: nv-gha-runners/setup-proxy-cache@main
         continue-on-error: true
 
+      - name: Install uv
+        uses: astral-sh/setup-uv@v7
+
       - name: Install dependencies
         uses: ./.github/actions/install_unix_deps
         continue-on-error: false
@@ -277,6 +280,11 @@ jobs:
           LOCAL_CTK: ${{ matrix.LOCAL_CTK }}
         run: run-tests core
 
+      - name: Run cuda.core examples with PEP 723 metadata in isolation
+        if: ${{ env.SKIP_CUDA_BINDINGS_TEST == '0' }}
+        run: |
+          ci/tools/run-examples-pep723 cuda_core/examples/
+
       - name: Ensure cuda-python installable
         run: |
           if [[ "${{ matrix.LOCAL_CTK }}" == 1 ]]; then

ci/tools/run-examples-pep723

@@ -0,0 +1,43 @@
+#!/bin/bash
+
+# SPDX-FileCopyrightText: Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+
+# Runs a directory of examples using `uv run` to test that the PEP 723
+# dependency metadata in them is valid.
+
+# We allow a resolution to fail for cuda_core or cuda_bindings only, since there
+# we assume it is pinned to some future release.  This allows examples to be
+# written for unreleased features.  Testing examples against prereleases is
+# already covered by CI elsewhere.
+
+TARGET_DIR=$1
+
+if [ -z "$TARGET_DIR" ]; then
+    echo "Usage: $0 <directory>"
+    exit 1
+fi
+
+FAILED=0
+
+while IFS= read -r -d '' file; do
+    echo "Running $file..."
+    OUTPUT=$(uv run "$file" 2>&1)
+    EXIT_CODE=$?
+
+    if [ $EXIT_CODE -ne 0 ]; then
+        if echo "$OUTPUT" | grep -Eq "Because only cuda-(core|bindings)"; then
+            echo "SKIPPED: $file (No solution found)"
+        else
+            echo "FAILED: $file"
+            echo "$OUTPUT"
+            FAILED=1
+        fi
+    else
+        echo "PASSED: $file"
+    fi
+done < <(find "$TARGET_DIR" -name "*.py" -print0)
+
+if [ $FAILED -ne 0 ]; then
+    exit 1
+fi

cuda_core/examples/cuda_graphs.py

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 #
 # SPDX-License-Identifier: Apache-2.0
 
@@ -10,6 +10,10 @@
 #
 # ################################################################################
 
+# /// script
+# dependencies = ["cuda_bindings", "cuda_core", "cupy"]
+# ///
+
 import sys
 import time
 

cuda_core/examples/gl_interop_plasma.py

@@ -53,9 +53,10 @@
 # effect popular in the demoscene). The window title shows the current FPS.
 # Close the window or press Escape to exit.
 #
-# Requirements
-# ============
-#   pip install pyglet
+
+# /// script
+# dependencies = ["cuda_bindings", "cuda_core>0.6.0", "pyglet"]
+# ///
 
 import ctypes
 import sys

cuda_core/examples/jit_lto_fractal.py

@@ -1,4 +1,4 @@
-# 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: Apache-2.0
 
@@ -12,6 +12,10 @@
 #
 # ################################################################################
 
+# /// script
+# dependencies = ["cuda_bindings", "cuda_core", "cupy"]
+# ///
+
 import argparse
 import sys
 

cuda_core/examples/memory_ops.py

@@ -1,4 +1,4 @@
-# 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: Apache-2.0
 
@@ -10,6 +10,10 @@
 #
 # ################################################################################
 
+# /// script
+# dependencies = ["cuda_bindings", "cuda_core", "cupy", "numpy"]
+# ///
+
 import sys
 
 import cupy as cp

cuda_core/examples/pytorch_example.py

@@ -1,4 +1,4 @@
-# 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: Apache-2.0
 
@@ -9,6 +9,10 @@
 #
 # ################################################################################
 
+# /// script
+# dependencies = ["cuda_bindings", "cuda_core", "torch"]
+# ///
+
 import sys
 
 import torch

cuda_core/examples/saxpy.py

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 #
 # SPDX-License-Identifier: Apache-2.0
 
@@ -10,6 +10,10 @@
 #
 # ################################################################################
 
+# /// script
+# dependencies = ["cuda_bindings", "cuda_core", "cupy"]
+# ///
+
 import sys
 
 import cupy as cp

cuda_core/examples/show_device_properties.py

@@ -1,4 +1,4 @@
-# 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: Apache-2.0
 
@@ -9,6 +9,10 @@
 #
 # ################################################################################
 
+# /// script
+# dependencies = ["cuda_bindings", "cuda_core"]
+# ///
+
 import sys
 
 from cuda.core import Device, system

cuda_core/examples/simple_multi_gpu_example.py

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 #
 # SPDX-License-Identifier: Apache-2.0
 
@@ -9,6 +9,10 @@
 #
 # ################################################################################
 
+# /// script
+# dependencies = ["cuda_bindings", "cuda_core", "cupy"]
+# ///
+
 import sys
 
 import cupy as cp