Skip to content

Commit 245b200

Browse files
committed
add ci for layerstest
1 parent e6ab64e commit 245b200

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

.github/workflows/ci-sharktank.yml

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -185,6 +185,21 @@ jobs:
185185
--iree-hal-target-device=hip \
186186
--iree-hip-target=gfx942 \
187187
--iree-device=hip://0 \
188+
--device=cuda:0 \
189+
--ignore=sharktank/tests/layers
190+
191+
- name: Run sharktank layers tests
192+
if: ${{ !cancelled() }}
193+
run: |
194+
pytest sharktank/tests/layers \
195+
--durations=10 \
196+
--capture=no \
197+
--log-cli-level=info \
198+
-v \
199+
--iree-hal-target-device=hip \
200+
--iree-hip-target=gfx942 \
201+
--iree-device=hip://0 \
202+
--parameters=/shark-dev/ossci-models/llama_3_1/405b/fp4/fp4_preshuffled_2025_09_12.irpa \
188203
--device=cuda:0
189204
190205
test_with_data:

0 commit comments

Comments
 (0)