Update nightly uv pipeline

.github/actions/bootstrap-cudnn-ci/action.yml

@@ -0,0 +1,52 @@
+name: Bootstrap cuDNN CI container
+description: Install OS dependencies and uv in CUDA cuDNN container jobs
+inputs:
+  python-version:
+    description: Python major.minor expected in the container
+    required: false
+    default: "3.12"
+runs:
+  using: composite
+  steps:
+    - name: Install system dependencies
+      shell: bash
+      run: |
+        set -euo pipefail
+        export DEBIAN_FRONTEND=noninteractive
+        apt-get update
+        apt-get install -y --no-install-recommends \
+          ca-certificates \
+          curl \
+          git \
+          gh \
+          build-essential \
+          cmake \
+          pkg-config \
+          python3 \
+          python3-dev \
+          python3-venv \
+          python3-pip \
+          zstd
+        ln -sf /usr/bin/python3 /usr/bin/python
+        rm -rf /var/lib/apt/lists/*
+
+    - name: Install uv
+      shell: bash
+      run: |
+        set -euo pipefail
+        curl -LsSf https://astral.sh/uv/install.sh | sh
+        echo "$HOME/.cargo/bin" >> "$GITHUB_PATH"
+
+    - name: Print toolchain versions
+      shell: bash
+      run: |
+        set -euo pipefail
+        python3 --version
+        uv --version
+        gcc --version | head -n 1
+        cmake --version | head -n 1
+        if command -v nvcc >/dev/null 2>&1; then
+          nvcc --version
+        else
+          echo "nvcc not found on PATH"
+        fi

.github/actions/setup-uv-env/action.yml

@@ -0,0 +1,83 @@
+name: Setup uv environment
+description: Restore uv and venv caches, and run uv sync on cache miss
+inputs:
+  uv-cache-key-prefix:
+    description: Prefix for uv package cache key
+    required: true
+  venv-cache-key-prefix:
+    description: Prefix for virtual environment cache key
+    required: true
+  venv-cache-key-suffix:
+    description: Deterministic suffix appended to venv cache key prefix
+    required: true
+  uv-cache-key-suffix:
+    description: Suffix for uv cache key (e.g. "latest" for a static key)
+    required: true
+outputs:
+  uv_cache_hit:
+    description: Whether uv package cache had an exact key hit
+    value: ${{ steps.restore-uv-cache.outputs.cache-hit }}
+  venv_cache_hit:
+    description: Whether venv cache had an exact key hit
+    value: ${{ steps.restore-venv-cache.outputs.cache-hit }}
+runs:
+  using: composite
+  steps:
+    - name: Restore uv package cache
+      id: restore-uv-cache
+      uses: actions/cache/restore@v4
+      with:
+        path: ~/.cache/uv
+        key: ${{ inputs.uv-cache-key-prefix }}-${{ inputs.uv-cache-key-suffix }}
+        fail-on-cache-miss: false
+        restore-keys: |
+          ${{ inputs.uv-cache-key-prefix }}-
+
+    - name: Restore venv cache
+      id: restore-venv-cache
+      uses: actions/cache/restore@v4
+      with:
+        path: .venv
+        key: ${{ inputs.venv-cache-key-prefix }}-${{ inputs.venv-cache-key-suffix }}
+        fail-on-cache-miss: false
+        restore-keys: |
+          ${{ inputs.venv-cache-key-prefix }}-
+
+    - name: Debug cache and environment context
+      shell: bash
+      run: |
+        set -euo pipefail
+        echo "::group::setup-uv-env debug context"
+        echo "uv cache key: ${{ inputs.uv-cache-key-prefix }}-${{ inputs.uv-cache-key-suffix }}"
+        echo "venv cache key: ${{ inputs.venv-cache-key-prefix }}-${{ inputs.venv-cache-key-suffix }}"
+        echo "uv cache exact hit: ${{ steps.restore-uv-cache.outputs.cache-hit }}"
+        echo "venv cache exact hit: ${{ steps.restore-venv-cache.outputs.cache-hit }}"
+        echo "workspace: $GITHUB_WORKSPACE"
+        df -h
+        echo "::endgroup::"
+
+    - name: Install dependencies with uv (dev + cu12)
+      if: steps.restore-venv-cache.outputs.cache-hit != 'true'
+      shell: bash
+      run: |
+        set -euo pipefail
+        export UV_LINK_MODE=copy
+        echo "::group::uv sync (dev + cu12)"
+        uv sync \
+          --frozen \
+          --group dev \
+          --extra cu12
+        echo "::endgroup::"
+        uv run python -c "import torch; print(f'torch={torch.__version__} cuda={torch.version.cuda}')"
+
+    - name: Report cache sizes
+      shell: bash
+      run: |
+        echo "::group::cache sizes"
+        echo "uv package cache:"
+        du -sh ~/.cache/uv 2>/dev/null || echo "  (not present)"
+        echo ".venv:"
+        du -sh .venv 2>/dev/null || echo "  (not present)"
+        df -h
+        echo "::endgroup::"
+