minor: add mla fp8 test (#1494)

2024-09-23 20:40:17 +08:00
parent e4780cf839
commit 42a2d82ba7
3 changed files with 52 additions and 0 deletions
--- a/.github/workflows/pr-test.yml
+++ b/.github/workflows/pr-test.yml
@@ -239,6 +239,7 @@ jobs:
        run: |
          cd test/srt
          python3 test_mla.py
+          python3 test_mla_fp8.py

      - name: Evaluate Data Parallelism Accuracy (TP=2)
        timeout-minutes: 10