fix typo (#2487)

2024-12-15 13:44:55 +08:00
parent e04d3f2897
commit e8dbdf75bc
1 changed files with 2 additions and 2 deletions
--- a/sgl-kernel/tests/test_trt_reduce.py
+++ b/sgl-kernel/tests/test_trt_reduce.py
@@ -236,8 +236,8 @@ class TestCustomAllReduce(unittest.TestCase):
            if rank == 0:
                logger.warning(
                    f"test_size = {sz}, world_size = {world_size}, "
-                    f"vllm time = {elapse_vllm * 1000 / test_loop:.4f}us,"
-                    f"custom time = {elapse_custom * 1000 / test_loop:.4f}us"
+                    f"vllm time = {elapse_vllm * 1000 / test_loop:.4f}ms,"
+                    f"custom time = {elapse_custom * 1000 / test_loop:.4f}ms"
                )

        self.free_custom_allreduce(group)