Skip to content

Commit

Permalink
[CI] Move punet integration test to ci-sharktank.yml
Browse files Browse the repository at this point in the history
  • Loading branch information
nithinsubbiah authored and dan-garvey committed Nov 26, 2024
1 parent ea907f3 commit 6257639
Showing 1 changed file with 25 additions and 0 deletions.
25 changes: 25 additions & 0 deletions .github/workflows/ci-sharktank.yml
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,31 @@ concurrency:
cancel-in-progress: true

jobs:
test_punet:
name: "Integration Tests - punet"
runs-on: nodai-amdgpu-mi250-x86-64
env:
VENV_DIR: ${{ github.workspace }}/.venv
steps:
- name: "Checkout Code"
uses: actions/checkout@v3

- name: "Setup Python venv"
run: python3 -m venv ${VENV_DIR}

- name: Install pip deps
run: |
source ${VENV_DIR}/bin/activate
python -m pip install --no-compile --upgrade pip
pip install --no-compile -r pytorch-rocm-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=iree-turbine"
pip install --no-compile -r requirements.txt -r sharktank/requirements-tests.txt -e sharktank/ shortfin/
- name: Run punet tests
run: |
source ${VENV_DIR}/bin/activate
pytest -v sharktank/ -m model_punet
test:
name: "Unit Tests and Type Checking"
strategy:
Expand Down

0 comments on commit 6257639

Please sign in to comment.