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

2025-04-11 02:28:41 +08:00
parent d050df368c
commit ed01b4515e
8 changed files with 63 additions and 76 deletions
--- a/sgl-kernel/tests/test_rotary_embedding.py
+++ b/sgl-kernel/tests/test_rotary_embedding.py
@@ -187,9 +187,6 @@ def test_correctness(
        pos_ids, query_flashinfer, key_flashinfer
    )

-    print(query_ref_out)
-    print(query_flashinfer_out)
-
    torch.testing.assert_close(
        query_ref_out, query_flashinfer_out, atol=1e-2, rtol=1e-2
    )