diff --git a/.github/workflows/ci-tk.yaml b/.github/workflows/ci-tk.yaml index 547deeb5..c935d488 100644 --- 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()"