Fix artifact name mismatch in nightly CI by passing source SHA

Artifact names embed the commit SHA from the build that created them.
When the nightly workflow downloads artifacts from a different CI run,
it must use that run's SHA — not github.sha (the nightly run's own
SHA) — to construct the correct artifact names.

- ci-nightly.yml: resolve head_sha from the source CI run via
  `gh run view --json headSha`, pass it to test workflows
- test-wheel-linux/windows.yml: add `sha` input (defaults to
  github.sha for backward compatibility), use it in env-vars

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: leofang

.github/workflows/ci-nightly.yml

@@ -36,6 +36,7 @@ jobs:
     runs-on: ubuntu-latest
     outputs:
       RUN_ID: ${{ steps.find.outputs.run_id }}
+      HEAD_SHA: ${{ steps.find.outputs.head_sha }}
       CUDA_BUILD_VER: ${{ steps.get-vars.outputs.cuda_build_ver }}
     steps:
       - name: Checkout repository
@@ -55,27 +56,34 @@ jobs:
           GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         run: |
           if [[ -n "${{ inputs.run-id }}" ]]; then
-            echo "run_id=${{ inputs.run-id }}" >> $GITHUB_OUTPUT
-            echo "Using manually specified run ID: ${{ inputs.run-id }}"
-            exit 0
+            RUN_ID="${{ inputs.run-id }}"
+            echo "Using manually specified run ID: $RUN_ID"
+          else
+            RUN_ID=$(gh run list \
+              -b main \
+              -L 1 \
+              -w "CI" \
+              -s success \
+              -R "${{ github.repository }}" \
+              --json databaseId \
+              | jq -r '.[0].databaseId')
+
+            if [[ -z "$RUN_ID" || "$RUN_ID" == "null" ]]; then
+              echo "::error::No successful CI run found on main"
+              exit 1
+            fi
+            echo "Using latest successful CI run: $RUN_ID"
           fi
 
-          RUN_ID=$(gh run list \
-            -b main \
-            -L 1 \
-            -w "CI" \
-            -s success \
+          # Resolve the head SHA from the CI run — artifact names embed this.
+          HEAD_SHA=$(gh run view "$RUN_ID" \
             -R "${{ github.repository }}" \
-            --json databaseId \
-            | jq -r '.[0].databaseId')
-
-          if [[ -z "$RUN_ID" || "$RUN_ID" == "null" ]]; then
-            echo "::error::No successful CI run found on main"
-            exit 1
-          fi
+            --json headSha \
+            | jq -r '.headSha')
 
           echo "run_id=$RUN_ID" >> $GITHUB_OUTPUT
-          echo "Using latest successful CI run: $RUN_ID"
+          echo "head_sha=$HEAD_SHA" >> $GITHUB_OUTPUT
+          echo "Source commit: $HEAD_SHA"
 
   # ── PyTorch interop tests ──
 
@@ -93,6 +101,7 @@ jobs:
       host-platform: linux-64
       build-ctk-ver: ${{ needs.find-wheels.outputs.CUDA_BUILD_VER }}
       run-id: ${{ needs.find-wheels.outputs.RUN_ID }}
+      sha: ${{ needs.find-wheels.outputs.HEAD_SHA }}
       test-mode: nightly-pytorch
       matrix_filter: 'map(select(.MODE == "nightly-pytorch"))'
 
@@ -110,6 +119,7 @@ jobs:
       host-platform: win-64
       build-ctk-ver: ${{ needs.find-wheels.outputs.CUDA_BUILD_VER }}
       run-id: ${{ needs.find-wheels.outputs.RUN_ID }}
+      sha: ${{ needs.find-wheels.outputs.HEAD_SHA }}
       test-mode: nightly-pytorch
       matrix_filter: 'map(select(.MODE == "nightly-pytorch"))'
 
@@ -129,6 +139,7 @@ jobs:
       host-platform: linux-64
       build-ctk-ver: ${{ needs.find-wheels.outputs.CUDA_BUILD_VER }}
       run-id: ${{ needs.find-wheels.outputs.RUN_ID }}
+      sha: ${{ needs.find-wheels.outputs.HEAD_SHA }}
       test-mode: nightly-numba-cuda
       matrix_filter: 'map(select(.MODE == "nightly-numba-cuda"))'
 
@@ -146,6 +157,7 @@ jobs:
       host-platform: linux-aarch64
       build-ctk-ver: ${{ needs.find-wheels.outputs.CUDA_BUILD_VER }}
       run-id: ${{ needs.find-wheels.outputs.RUN_ID }}
+      sha: ${{ needs.find-wheels.outputs.HEAD_SHA }}
       test-mode: nightly-numba-cuda
       matrix_filter: 'map(select(.MODE == "nightly-numba-cuda"))'
 
@@ -163,6 +175,7 @@ jobs:
       host-platform: win-64
       build-ctk-ver: ${{ needs.find-wheels.outputs.CUDA_BUILD_VER }}
       run-id: ${{ needs.find-wheels.outputs.RUN_ID }}
+      sha: ${{ needs.find-wheels.outputs.HEAD_SHA }}
       test-mode: nightly-numba-cuda
       matrix_filter: 'map(select(.MODE == "nightly-numba-cuda"))'
 

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

@@ -41,6 +41,12 @@ on:
           'nightly-numba-cuda'.
         type: string
         default: 'standard'
+      sha:
+        description: >
+          Commit SHA used to construct artifact names.
+          Defaults to github.sha (current run) when empty.
+        type: string
+        default: ''
 
 defaults:
   run:
@@ -131,7 +137,7 @@ jobs:
           HOST_PLATFORM: ${{ inputs.host-platform }}
           LOCAL_CTK: ${{ matrix.LOCAL_CTK }}
           PY_VER: ${{ matrix.PY_VER }}
-          SHA: ${{ github.sha }}
+          SHA: ${{ inputs.sha || github.sha }}
           SKIP_BINDINGS_TEST_OVERRIDE: ${{ inputs.skip-bindings-test && '1' || '0' }}
         run: ./ci/tools/env-vars test
 

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

@@ -41,6 +41,12 @@ on:
           'nightly-numba-cuda'.
         type: string
         default: 'standard'
+      sha:
+        description: >
+          Commit SHA used to construct artifact names.
+          Defaults to github.sha (current run) when empty.
+        type: string
+        default: ''
 
 jobs:
   compute-matrix:
@@ -125,7 +131,7 @@ jobs:
           HOST_PLATFORM: ${{ inputs.host-platform }}
           LOCAL_CTK: ${{ matrix.LOCAL_CTK }}
           PY_VER: ${{ matrix.PY_VER }}
-          SHA: ${{ github.sha }}
+          SHA: ${{ inputs.sha || github.sha }}
           SKIP_BINDINGS_TEST_OVERRIDE: ${{ inputs.skip-bindings-test && '1' || '0' }}
         shell: bash --noprofile --norc -xeuo pipefail {0}
         run: ./ci/tools/env-vars test