fix typo (#2487)

e8dbdf75 · Yineng Zhang · GitHub · e04d3f28 · e8dbdf75
Unverified Commit e8dbdf75 authored Dec 15, 2024 by Yineng Zhang Committed by GitHub Dec 15, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

sgl-kernel/tests/test_trt_reduce.py sgl-kernel/tests/test_trt_reduce.py +2 -2

No files found.
--- 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)