From 5eb013d878093cab5f92315006cd9fee526f6e5a Mon Sep 17 00:00:00 2001 From: saienduri <77521230+saienduri@users.noreply.github.com> Date: Tue, 20 Aug 2024 01:22:28 -0700 Subject: [PATCH] add caching --- .github/workflows/test_models.yml | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/.github/workflows/test_models.yml b/.github/workflows/test_models.yml index 03872dea3..94b9ca1bf 100644 --- a/.github/workflows/test_models.yml +++ b/.github/workflows/test_models.yml @@ -23,6 +23,14 @@ jobs: os: [nodai-amdgpu-mi250-x86-64] runs-on: ${{matrix.os}} + env: + E2E_VENV_DIR: ${{ github.workspace }}/test-suite_venv + HF_TOKEN: ${{ secrets.HF_TOKEN }} + AZ_PRIVATE_CONNECTION: ${{ secrets.ONNXPRIVATESTORAGE_AZ_PRIVATE_CONNECTION }} + ROCR_VISIBLE_DEVICES: ${{ matrix.visible-device }} + TORCH_HOME: /groups/aig_sharks/test-suite-ci-cache + HF_HOME: /groups/aig_sharks/test-suite-ci-cache + TURBINE_TANK_CACHE_DIR: /groups/aig_sharks/test-suite-ci-cache steps: # We are using a persistent Gentoo runner here, and this python action is not supported for the arch # - name: "Setting up Python" @@ -73,4 +81,4 @@ jobs: pytest -v models/turbine_models/tests/sdxl_test.py --device vulkan --rt_device vulkan --iree_target_triple rdna3-unknown-linux pytest -v models/turbine_models/tests/sdxl_test.py --device rocm --rt_device hip --iree_target_triple gfx90a --precision fp16 --attn_spec default pytest -v models/turbine_models/tests/sdxl_test.py --device rocm --rt_device hip --iree_target_triple gfx90a --precision fp16 --attn_spec default --batch_size 2 - pytest -v models/turbine_models/tests/sd3_test.py --device cpu --rt_device local-task --iree_target_triple x86_64-linux-gnu --num_inference_steps 5 \ No newline at end of file + pytest -v models/turbine_models/tests/sd3_test.py --device cpu --rt_device local-task --iree_target_triple x86_64-linux-gnu --num_inference_steps 5