Move fa4 from sgl-kernel to jit kernel (#17353)

2026-01-24 15:25:03 +08:00
parent 7a4bb0d516
commit 3992a023e6
45 changed files with 24594 additions and 11 deletions
--- a/.github/workflows/pr-test.yml
+++ b/.github/workflows/pr-test.yml
@@ -1241,6 +1241,11 @@ jobs:
          fi
          IS_BLACKWELL=1 python3 run_suite.py --hw cuda --suite stage-b-test-4-gpu-b200 $CONTINUE_ON_ERROR_FLAG

+      - name: Run FA4 jit_kernel tests (SM100+)
+        timeout-minutes: 5
+        run: |
+          IS_BLACKWELL=1 python3 -m pytest -q python/sglang/jit_kernel/tests/test_flash_attention_4.py
+
  stage-c-test-large-4-gpu:
    needs: [check-changes, call-gate, wait-for-stage-b, sgl-kernel-build-wheels]
    if: |