16

Signed-off-by: Ivan Butygin <ivan.butygin@gmail.com>
iree-org · Dec 20, 2024 · 0004ae3 · 0004ae3
1 parent 8d5d130
commit 0004ae3
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=8
-          WAVE_CACHE_ON=0 pytest -n 8 --capture=tee-sys -vv ./tests/kernel/wave/
+          export WAVE_DISTRIBUTE_GPU_TESTS=16
+          WAVE_CACHE_ON=0 pytest -n 16 --capture=tee-sys -vv ./tests/kernel/wave/
 
       - name: Run e2e tests on AMD GPU MI250
         if: "contains(matrix.os, 'mi250') && !cancelled()"