Skip to content

Commit b9eeea7

Browse files
committed
Update CUDA version in workflow to match CI runner
The Windows GPU runners have CUDA 13.0 installed. The export job continues to use 12.8 (via Docker), relying on CUDA forward compatibility for the AOTI blobs.
1 parent 47c949d commit b9eeea7

File tree

1 file changed

+4
-10
lines changed

1 file changed

+4
-10
lines changed

.github/workflows/cuda-windows.yml

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,6 @@ concurrency:
1515
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}-${{ github.event_name == 'workflow_dispatch' }}-${{ github.event_name == 'schedule' }}
1616
cancel-in-progress: false
1717

18-
env:
19-
CUDA_VERSION: "12.8"
20-
2118
jobs:
2219
export-model-cuda-windows-artifact:
2320
name: export-model-cuda-windows-artifact
@@ -44,7 +41,7 @@ jobs:
4441
secrets-env: EXECUTORCH_HF_TOKEN
4542
runner: linux.g5.4xlarge.nvidia.gpu
4643
gpu-arch-type: cuda
47-
gpu-arch-version: ${{ env.CUDA_VERSION }}
44+
gpu-arch-version: 12.8
4845
docker-image: ci-image:executorch-ubuntu-22.04-cuda-windows
4946
submodules: recursive
5047
upload-artifact: ${{ matrix.model.repo }}-${{ matrix.model.name }}-cuda-windows-${{ matrix.quant }}
@@ -101,6 +98,8 @@ jobs:
10198
- "quantized-int4-weight-only"
10299
with:
103100
timeout: 240
101+
gpu-arch-type: cuda
102+
gpu-arch-version: 13.0
104103
submodules: recursive
105104
download-artifact: ${{ matrix.model.repo }}-${{ matrix.model.name }}-cuda-windows-${{ matrix.quant }}
106105
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
@@ -117,10 +116,5 @@ jobs:
117116
throw 'RUNNER_ARTIFACT_DIR is empty. Ensure download-artifact is configured for windows_job.yml.'
118117
}
119118
120-
.ci/scripts/test_model_e2e_windows.ps1 `
121-
-Device cuda-windows `
122-
-HfModel '${{ matrix.model.repo }}/${{ matrix.model.name }}' `
123-
-QuantName '${{ matrix.quant }}' `
124-
-ModelDir \$artifactDir `
125-
-ExpectedCudaVersion '${{ env.CUDA_VERSION }}'
119+
.ci/scripts/test_model_e2e_windows.ps1 -Device cuda-windows -HfModel '${{ matrix.model.repo }}/${{ matrix.model.name }}' -QuantName '${{ matrix.quant }}' -ModelDir \$artifactDir -ExpectedCudaVersion '13.0'
126120
}"

0 commit comments

Comments
 (0)