[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_fp8_blockwise_gemm.py
+++ b/sgl-kernel/tests/test_fp8_blockwise_gemm.py
@@ -79,7 +79,6 @@ def _test_accuracy_once(M, N, K, out_dtype, device):
    rtol = 0.02
    atol = 1
    torch.testing.assert_close(o, o1, rtol=rtol, atol=atol)
-    print(f"M={M}, N={N}, K={K}, out_dtype={out_dtype}: OK")


@pytest.mark.parametrize("M", [1, 3, 5, 127, 128, 512, 1024, 4096])