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

[Misc] Clean sgl-kernel test (#5216)

Browse Source
This commit is contained in:
PGFLMG
2025-04-11 02:28:41 +08:00
committed by GitHub
parent d050df368c
commit ed01b4515e
8 changed files with 63 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -70,8 +70,6 @@ def test_lightning_attention_decode(dtype, batch_size, num_heads, dim, embed_dim
ref_output,
rtol=rtol,
atol=atol,
msg=f"Output mismatch for batch_size={batch_size}, num_heads={num_heads}, "
f"dim={dim}, embed_dim={embed_dim}, dtype={dtype}",
)
torch.testing.assert_close(
@@ -79,8 +77,6 @@ def test_lightning_attention_decode(dtype, batch_size, num_heads, dim, embed_dim
ref_new_kv,
rtol=rtol,
atol=atol,
msg=f"New KV mismatch for batch_size={batch_size}, num_heads={num_heads}, "
f"dim={dim}, embed_dim={embed_dim}, dtype={dtype}",
)