EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Add main for merge state tests (#6492)

Browse Source 
Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
This commit is contained in:
Yuan Luo
2025-05-22 12:56:25 +08:00
committed by GitHub
parent 3bde101099
commit 121f92c583
2 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sgl-kernel/tests/test_merge_state.py
View File

@@ -136,3 +136,7 @@ def test_merge_state(seq_len, num_heads, head_dim):
assert torch.allclose(v_merged, v_merged_std, atol=1e-2) assert torch.allclose(v_merged, v_merged_std, atol=1e-2)
assert torch.allclose(s_merged, s_merged_std, atol=1e-2) assert torch.allclose(s_merged, s_merged_std, atol=1e-2)
if __name__ == "__main__":
pytest.main([__file__])

4
sgl-kernel/tests/test_merge_state_v2.py
View File

@@ -394,3 +394,7 @@ def test_merge_attn_states(
len(NUM_BATCH_TOKENS) * len(HEAD_SIZES) * len(NUM_QUERY_HEADS) * len(DTYPES) len(NUM_BATCH_TOKENS) * len(HEAD_SIZES) * len(NUM_QUERY_HEADS) * len(DTYPES)
): ):
generate_markdown_table() generate_markdown_table()
if __name__ == "__main__":
pytest.main([__file__])