Add accuracy test for TP torch compile (#3994)

2025-03-03 05:18:18 +08:00
parent 00ce7e311c
commit d3fe9bae56
2 changed files with 71 additions and 0 deletions
--- a/.github/workflows/pr-test.yml
+++ b/.github/workflows/pr-test.yml
@@ -161,6 +161,12 @@ jobs:
          cd test/srt
          python3 test_moe_ep.py

+      - name: Test torch compile (TP=2)
+        timeout-minutes: 10
+        run: |
+          cd test/srt
+          python3 test_mla_tp.py
+
  performance-test-1-gpu-part-1:
    needs: filter
    if: (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') &&