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

2025-01-23 21:13:55 +08:00
parent 07a22cbba3
commit 0da0989ad4
6 changed files with 13 additions and 4 deletions
--- a/sgl-kernel/tests/test_activation.py
+++ b/sgl-kernel/tests/test_activation.py
@@ -35,4 +35,5 @@ def test_fused_gelu_mul(dim, batch_size, seq_len):
    torch.testing.assert_close(y_ref, y, rtol=1e-3, atol=1e-3)


-test_fused_silu_mul(128, 1, 1)
+if __name__ == "__main__":
+    pytest.main([__file__])