Skip to content

Commit 6b56c5a

Browse files
authored
Update CUDA version in workflow to 12.8
1 parent 47c949d commit 6b56c5a

1 file changed

Lines changed: 4 additions & 5 deletions

File tree

.github/workflows/cuda-windows.yml

Lines changed: 4 additions & 5 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: "12.8"
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 }}
@@ -122,5 +121,5 @@ jobs:
122121
-HfModel '${{ matrix.model.repo }}/${{ matrix.model.name }}' `
123122
-QuantName '${{ matrix.quant }}' `
124123
-ModelDir \$artifactDir `
125-
-ExpectedCudaVersion '${{ env.CUDA_VERSION }}'
124+
-ExpectedCudaVersion '12.8'
126125
}"

0 commit comments

Comments
 (0)