File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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-
2118jobs :
2219 export-model-cuda-windows-artifact :
2320 name : export-model-cuda-windows-artifact
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 }}
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 }"
You can’t perform that action at this time.
0 commit comments