fix more

pytorch · May 17, 2024 · 7514aae · 7514aae
1 parent 3600f38
commit 7514aae
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/.github/workflows/_test.yml b/.github/workflows/_test.yml
@@ -92,6 +92,13 @@ jobs:
         with:
           name: torch-xla-wheels
           path: /tmp/wheels/
+        if: ! ${{ inputs.require-pytorch-cuda }}
+      - name: Fetch wheels (pytorch and pytorch/xla with CUDA enabled) if needed.
+        uses: actions/download-artifact@v4
+        with:
+          name: torch-with-cuda-xla-with-cuda-wheels
+          path: /tmp/wheels/
+        if: ${{ inputs.require-pytorch-cuda }}
       - name: Fetch CPP test binaries
         uses: actions/download-artifact@v4
         with: