fix typo (#2487)

sgl-project · Dec 15, 2024 · e8dbdf7 · e8dbdf7
1 parent e04d3f2
commit e8dbdf7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sgl-kernel/tests/test_trt_reduce.py b/sgl-kernel/tests/test_trt_reduce.py
@@ -236,8 +236,8 @@ def performance(self, world_size, rank, distributed_init_port):
             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)