Move test. Edit CI

nod-ai · Aug 22, 2024 · cf76f86 · cf76f86
1 parent 996c42b
commit cf76f86
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 784 deletions.
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -53,9 +53,11 @@ jobs:
           # from non default locations first. Installing the PyTorch CPU
           # wheels saves multiple minutes and a lot of bandwidth on runner setup.
           pip install --no-compile -r pytorch-cpu-requirements.txt
+          pip install --no-compile -r tuner-requirements.txt
           pip install --no-compile -f https://iree.dev/pip-release-links.html --src deps \
             -e "git+https://github.com/iree-org/iree-turbine.git#egg=shark-turbine"
           pip install --no-compile -r requirements.txt -e sharktank/ shortfin/
+          pip install --no-compile -f https://iree.dev/pip-release-links.html iree-compiler iree-runtime
 
       - name: Run sharktank tests
         if: ${{ !cancelled() }}