sync flashinfer and update sgl-kernel tests (#3081)

This commit is contained in:
Yineng Zhang
2025-01-23 21:13:55 +08:00
committed by GitHub
parent 07a22cbba3
commit 0da0989ad4
6 changed files with 13 additions and 4 deletions

View File

@@ -82,3 +82,7 @@ def test_lightning_attention_decode(dtype, batch_size, num_heads, dim, embed_dim
msg=f"New KV mismatch for batch_size={batch_size}, num_heads={num_heads}, "
f"dim={dim}, embed_dim={embed_dim}, dtype={dtype}",
)
if __name__ == "__main__":
pytest.main([__file__])