Revert "16"

This reverts commit 0004ae3. Signed-off-by: Ivan Butygin <[email protected]>
iree-org · Dec 20, 2024 · 4db5578 · 4db5578
1 parent 0004ae3
commit 4db5578
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/ci-tk.yaml b/.github/workflows/ci-tk.yaml
@@ -76,8 +76,8 @@ jobs:
         run: |
           pip install --no-compile -r pytorch-rocm-requirements.txt
           export WAVE_RUN_E2E_TESTS=1
-          export WAVE_DISTRIBUTE_GPU_TESTS=16
-          WAVE_CACHE_ON=0 pytest -n 16 --capture=tee-sys -vv ./tests/kernel/wave/
+          export WAVE_DISTRIBUTE_GPU_TESTS=8
+          WAVE_CACHE_ON=0 pytest -n 8 --capture=tee-sys -vv ./tests/kernel/wave/
 
       - name: Run e2e tests on AMD GPU MI250
         if: "contains(matrix.os, 'mi250') && !cancelled()"