fix typo (#2487)
This commit is contained in:
@@ -236,8 +236,8 @@ class TestCustomAllReduce(unittest.TestCase):
|
|||||||
if rank == 0:
|
if rank == 0:
|
||||||
logger.warning(
|
logger.warning(
|
||||||
f"test_size = {sz}, world_size = {world_size}, "
|
f"test_size = {sz}, world_size = {world_size}, "
|
||||||
f"vllm time = {elapse_vllm * 1000 / test_loop:.4f}us,"
|
f"vllm time = {elapse_vllm * 1000 / test_loop:.4f}ms,"
|
||||||
f"custom time = {elapse_custom * 1000 / test_loop:.4f}us"
|
f"custom time = {elapse_custom * 1000 / test_loop:.4f}ms"
|
||||||
)
|
)
|
||||||
|
|
||||||
self.free_custom_allreduce(group)
|
self.free_custom_allreduce(group)
|
||||||
|
|||||||
Reference in New Issue
Block a user