From e21e1f2f481031ef7f22dacdab76529851f1c38e Mon Sep 17 00:00:00 2001 From: Xuanwo Date: Tue, 27 Oct 2026 00:07:47 +0800 Subject: [PATCH] ci(python): exercise forked DataLoader tests with CPU PyTorch --- .github/workflows/python.yml | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/.github/workflows/python.yml b/.github/workflows/python.yml index 3608c24ca..5d85ee6b6 100644 --- a/.github/workflows/python.yml +++ b/.github/workflows/python.yml @@ -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