Add main for merge state tests (#6492)

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
2025-05-22 12:56:25 +08:00
parent 3bde101099
commit 121f92c583
2 changed files with 8 additions and 0 deletions
--- a/sgl-kernel/tests/test_merge_state_v2.py
+++ b/sgl-kernel/tests/test_merge_state_v2.py
@@ -394,3 +394,7 @@ def test_merge_attn_states(
        len(NUM_BATCH_TOKENS) * len(HEAD_SIZES) * len(NUM_QUERY_HEADS) * len(DTYPES)
    ):
        generate_markdown_table()
+
+
+if __name__ == "__main__":
+    pytest.main([__file__])