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

2025-04-10 11:28:41 -07:00
parent d050df368c
commit ed01b4515e
8 changed files with 63 additions and 76 deletions
@@ -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
    )