From e8dbdf75bc72c7e5ba0230ceebc253a2eaa1bd6d Mon Sep 17 00:00:00 2001 From: Yineng Zhang Date: Sun, 15 Dec 2024 13:44:55 +0800 Subject: [PATCH] fix typo (#2487) --- sgl-kernel/tests/test_trt_reduce.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sgl-kernel/tests/test_trt_reduce.py b/sgl-kernel/tests/test_trt_reduce.py index 09a119080..c16d6bb6f 100644 --- 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)