[Log] Only Print Profiler Results on Rank 0 (#23370)

Signed-off-by: yewentao256 <zhyanwentao@126.com>
2025-09-02 14:53:34 -04:00
parent 598bd74cf8
commit 98aee612aa
2 changed files with 8 additions and 4 deletions
--- a/vllm/v1/worker/gpu_worker.py
+++ b/vllm/v1/worker/gpu_worker.py
@@ -400,8 +400,10 @@ class Worker(WorkerBase):
            self.profiler.start()
        else:
            self.profiler.stop()
-            print(self.profiler.key_averages().table(
-                sort_by="self_cuda_time_total"))
+            # only print profiler results on rank 0
+            if self.local_rank == 0:
+                print(self.profiler.key_averages().table(
+                    sort_by="self_cuda_time_total"))

    def execute_dummy_batch(self) -> None:
        self.model_runner._dummy_run(1)