Skip to content

Commit 42d84b2

Browse files
committed
add ci for layerstest
1 parent 66e3e61 commit 42d84b2

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

.github/workflows/ci-sharktank.yml

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -187,6 +187,20 @@ jobs:
187187
--iree-device=hip://0 \
188188
--device=cuda:0
189189
190+
- name: Run sharktank layers tests
191+
if: ${{ !cancelled() }}
192+
run: |
193+
pytest sharktank/tests/layers \
194+
--durations=10 \
195+
--capture=no \
196+
--log-cli-level=info \
197+
-v \
198+
--iree-hal-target-device=hip \
199+
--iree-hip-target=gfx942 \
200+
--iree-device=hip://0 \
201+
--parameters=/shark-dev/ossci-models/llama_3_1/405b/fp4/fp4_preshuffled_2025_09_12.irpa \
202+
--device=cuda:0
203+
190204
test_with_data:
191205
name: "Data-dependent Tests"
192206
strategy:

0 commit comments

Comments
 (0)