Split test_mla.py into two files (#4216)

2025-03-08 15:40:49 -08:00
parent b3251e9f40
commit 48473684cc
4 changed files with 116 additions and 102 deletions
--- a/.github/workflows/pr-test-amd.yml
+++ b/.github/workflows/pr-test-amd.yml
@@ -90,11 +90,11 @@ jobs:
      - name: MLA TEST
        timeout-minutes: 20
        run: |
-          docker exec -w /sglang-checkout/test/srt ci_sglang python3 test_mla.py TestMLA
+          docker exec -w /sglang-checkout/test/srt ci_sglang python3 test_mla.py

  finish:
    needs: [
-      accuracy-test-1-gpu-amd
+      accuracy-test-1-gpu-amd, mla-test-1-gpu-amd
    ]
    runs-on: ubuntu-latest
    steps: