File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -80,7 +80,7 @@ case "${IMAGE_NAME}" in
8080 LINTRUNNER=" "
8181 GCC_VERSION=11
8282 CUDA_WINDOWS_CROSS_COMPILE=yes
83- CUDA_VERSION=13.0
83+ CUDA_VERSION=12.6
8484 SKIP_PYTORCH=yes
8585 ;;
8686 * )
Original file line number Diff line number Diff line change 6464 secrets-env : EXECUTORCH_HF_TOKEN
6565 runner : linux.g5.4xlarge.nvidia.gpu
6666 gpu-arch-type : cuda
67- gpu-arch-version : 13.0
67+ gpu-arch-version : 12.6
6868 docker-image : ci-image:executorch-ubuntu-22.04-cuda-windows
6969 submodules : recursive
7070 upload-artifact : ${{ matrix.model_repo }}-${{ matrix.model_name }}-cuda-windows-${{ matrix.quant }}
@@ -146,7 +146,7 @@ jobs:
146146 timeout : 240
147147 runner : windows.g5.4xlarge.nvidia.gpu
148148 gpu-arch-type : cuda
149- gpu-arch-version : 13.0
149+ gpu-arch-version : 12.6
150150 download-artifact : ${{ matrix.model_repo }}-${{ matrix.model_name }}-cuda-windows-${{ matrix.quant }}
151151 ref : ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
152152 script : |
@@ -158,7 +158,7 @@ jobs:
158158 \$ErrorActionPreference = 'Stop'
159159 \$PSNativeCommandUseErrorActionPreference = \$true
160160
161- \$env:CUDA_HOME = 'C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v13.0 '
161+ \$env:CUDA_HOME = 'C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6 '
162162 \$env:CUDA_PATH = \$env:CUDA_HOME
163163 \$env:PATH = \"\$env:CUDA_HOME\bin;\$env:PATH\"
164164 nvcc --version
@@ -169,5 +169,5 @@ jobs:
169169 throw 'RUNNER_ARTIFACT_DIR is empty. Ensure download-artifact is configured for windows_job.yml.'
170170 }
171171
172- .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 '
172+ .ci/scripts/test_model_e2e_windows.ps1 -Device cuda-windows -HfModel '${{ matrix.model_repo }}/${{ matrix.model_name }}' -QuantName '${{ matrix.quant }}' -ModelDir \$artifactDir -ExpectedCudaVersion '12.6 '
173173 }"
You can’t perform that action at this time.
0 commit comments