Skip to content

Commit

Permalink
Enable hip tests for the Sharktank CI (iree-org#19349)
Browse files Browse the repository at this point in the history
Switching this to run on a w7900 machine we can run presubmits on hip.

---------

Signed-off-by: Rob Suderman <rob.suderman@gmail.com>
  • Loading branch information
rsuderman authored Dec 10, 2024
1 parent eae7bfb commit eff0671
Showing 1 changed file with 11 additions and 5 deletions.
16 changes: 11 additions & 5 deletions .github/workflows/pkgci_test_sharktank.yml
Original file line number Diff line number Diff line change
Expand Up @@ -25,10 +25,16 @@ jobs:
fail-fast: false
matrix:
include:
# CPU
- name: cpu_llvm_task
- name: cpu_task
target: target_cpu
gpu: none
runs-on: ubuntu-20.04

- name: hip_task
target: target_hip
gpu: gfx1100
runs-on: nodai-amdgpu-w7900-x86-64

env:
VENV_DIR: ${{ github.workspace }}/venv
steps:
Expand All @@ -54,7 +60,7 @@ jobs:
uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
with:
repository: iree-org/iree-test-suites
ref: a0c84d59c4332463dd46a3c4877d8e0ab2e0a80d
ref: c644a9dfc3e5e1a9d071d5e786b79cf612e9b1d3
path: iree-test-suites
lfs: true
- name: Install Sharktank models test suite requirements
Expand All @@ -64,10 +70,10 @@ jobs:
- name: Run Sharktank models test suite
run: |
source ${VENV_DIR}/bin/activate
pytest iree-test-suites/sharktank_models/ \
HIP_TARGET=${{ matrix.gpu }} pytest iree-test-suites/sharktank_models/ \
-rA \
-m "target_cpu" \
--log-cli-level=info \
--override-ini=xfail_strict=false \
-m ${{ matrix.target }} \
--timeout=120 \
--durations=0

0 comments on commit eff0671

Please sign in to comment.