Fix triton head num (#1482)

2024-09-21 10:25:20 +08:00
parent 014982b5e0
commit a68cb201dd
4 changed files with 54 additions and 1 deletions
--- a/.github/workflows/pr-test.yml
+++ b/.github/workflows/pr-test.yml
@@ -233,6 +233,12 @@ jobs:
        run: |
          cd test/srt
          python3 test_moe_eval_accuracy_large.py
+      
+      - name: Evaluate MLA Accuracy (TP=2)
+        timeout-minutes: 10
+        run: |
+          cd test/srt
+          python3 -m unittest test_mla.TestMLA.test_mmlu

  finish:
    needs: [