From 6904a457c558bdf8bec9a8de4bc2084254888571 Mon Sep 17 00:00:00 2001 From: saienduri Date: Thu, 14 Mar 2024 03:10:37 -0700 Subject: [PATCH] support for new step --- .github/workflows/test_e2eshark.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/test_e2eshark.yml b/.github/workflows/test_e2eshark.yml index 9f26f18cd..61e38e9de 100644 --- a/.github/workflows/test_e2eshark.yml +++ b/.github/workflows/test_e2eshark.yml @@ -132,7 +132,7 @@ jobs: path: turbine e2eshark: - needs: [turbine] + needs: [turbine, torch-mlir] strategy: matrix: version: [3.11] @@ -170,7 +170,7 @@ jobs: - name: Run Tests run: | + source ${E2E_VENV_DIR}/bin/activate cd e2eshark python ./run.py -c ../../torch-mlir/build --report --cachedir ~/.cache/huggingface --mode turbine --tests pytorch/models/beit-base-patch16-224-pt22k-ft22k pytorch/models/bge-base-en-v1.5 pytorch/models/mit-b0 pytorch/models/bert-large-uncased pytorch/models/deit-small-distilled-patch16-224 --cleanup - deactivate working-directory: ./test-suite