Enable `hip` tests for the Sharktank CI (#19349)
Switching this to run on a w7900 machine we can run presubmits on hip.
---------
Signed-off-by: Rob Suderman <rob.suderman@gmail.com>
diff --git a/.github/workflows/pkgci_test_sharktank.yml b/.github/workflows/pkgci_test_sharktank.yml
index c58216d..509b59e 100644
--- a/.github/workflows/pkgci_test_sharktank.yml
+++ b/.github/workflows/pkgci_test_sharktank.yml
@@ -25,10 +25,16 @@
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:
@@ -54,7 +60,7 @@
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
@@ -64,10 +70,10 @@
- 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