ci(python): exercise forked DataLoader tests with CPU PyTorch

2026-10-27 14:06:14 +00:00 · 2026-10-27 00:07:47 +08:00
parent 6bfd5ef0c6
commit e21e1f2f48
1 changed files with 4 additions and 0 deletions
@@ -151,6 +151,10 @@ jobs:
      - uses: ./.github/workflows/build_linux_wheel
        with:
          args: --profile ci
+      - name: Install CPU-only PyTorch
+        if: ${{ matrix.python-minor-version == '10' }}
+        run: |
+          python -m pip install 'torch>=2.0.0' --index-url https://download.pytorch.org/whl/cpu
      - uses: ./.github/workflows/run_tests
        with:
          integration: true