From 8c661def03339f7c5998a32538cbe8a165499c47 Mon Sep 17 00:00:00 2001 From: Ivan Butygin Date: Fri, 10 Jan 2025 16:40:31 +0100 Subject: [PATCH] [TKW] CI: Install GPU deps in separate step Signed-off-by: Ivan Butygin --- .github/workflows/ci-tk.yaml | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/.github/workflows/ci-tk.yaml b/.github/workflows/ci-tk.yaml index a9a4d060f..73bfe8407 100644 --- a/.github/workflows/ci-tk.yaml +++ b/.github/workflows/ci-tk.yaml @@ -63,10 +63,14 @@ jobs: run: | pytest -n 4 --capture=tee-sys -vv ./tests/kernel/wave/ - - name: Test TKW runtime related stack on amdgpu + - name: Install GPU deps if: "contains(matrix.os, 'amdgpu') && !cancelled()" run: | pip install --no-compile -r pytorch-rocm-requirements.txt + + - name: Test TKW runtime related stack on amdgpu + if: "contains(matrix.os, 'amdgpu') && !cancelled()" + run: | export export WAVE_CACHE_DIR=$PWD/.wave rm -rf ./.wave WAVE_CACHE_ON=1 pytest --capture=tee-sys -vv --run-e2e ./tests/kernel/wave/runtime @@ -74,13 +78,11 @@ jobs: - name: Run e2e tests on AMD GPU MI300 if: "contains(matrix.os, 'mi300') && !cancelled()" run: | - pip install --no-compile -r pytorch-rocm-requirements.txt WAVE_CACHE_ON=0 pytest -n 8 --capture=tee-sys -vv --run-e2e --gpu-distribute 8 ./tests/kernel/wave/ - name: Run e2e tests on AMD GPU MI250 if: "contains(matrix.os, 'mi250') && !cancelled()" run: | - pip install --no-compile -r pytorch-rocm-requirements.txt WAVE_CACHE_ON=0 pytest -n 2 --capture=tee-sys --run-e2e -vv ./tests/kernel/wave/ - name: Run LIT tests