Update validate-aarch64-linux-binaries.yml

pytorch · Sep 3, 2024 · 2feb328 · 2feb328
1 parent 543e421
commit 2feb328
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/.github/workflows/validate-aarch64-linux-binaries.yml b/.github/workflows/validate-aarch64-linux-binaries.yml
@@ -137,6 +137,10 @@ jobs:
           export AUDIO_RELEASE_VERSION=${{ fromJson(inputs.release-matrix).torchaudio }}
         fi
 
+        if [[ ${{ matrix.gpu_arch_version }} == "cuda-aarch64" ]]; then
+          export TORCH_ONLY=true
+        fi
+
         printf '%s\n' ${{ toJson(inputs.release-matrix) }} > release_matrix.json
         eval "$(conda shell.bash hook)"