File tree Expand file tree Collapse file tree 1 file changed +4
-10
lines changed
Expand file tree Collapse file tree 1 file changed +4
-10
lines changed 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 : 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 }"
You can’t perform that action at this time.
0 commit comments