From 48f85d464ff0b2e2aefec5a58c91f5e5ce9dce2c Mon Sep 17 00:00:00 2001 From: fzyzcjy <5236035+fzyzcjy@users.noreply.github.com> Date: Sat, 16 May 2026 09:24:39 +0800 Subject: [PATCH] Fix V2 trace filename collisions when DP/PP/EP enabled (#25446) --- python/sglang/srt/utils/profile_utils.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/python/sglang/srt/utils/profile_utils.py b/python/sglang/srt/utils/profile_utils.py index 79fa63c71..0b2fbeccc 100644 --- a/python/sglang/srt/utils/profile_utils.py +++ b/python/sglang/srt/utils/profile_utils.py @@ -293,12 +293,12 @@ class _ProfilerTorch(_ProfilerConcreteBase): filename_parts = [self.profile_id, f"TP-{self.ps.tp_rank}"] # Only add other ranks if parallelism is enabled (size > 1) - if getattr(self, "dp_size", 1) > 1: - filename_parts.append(f"DP-{getattr(self, 'dp_rank', 0)}") - if getattr(self, "pp_size", 1) > 1: - filename_parts.append(f"PP-{getattr(self, 'pp_rank', 0)}") - if getattr(self, "moe_ep_size", 1) > 1: - filename_parts.append(f"EP-{getattr(self, 'moe_ep_rank', 0)}") + if self.ps.dp_size > 1: + filename_parts.append(f"DP-{self.ps.dp_rank}") + if self.ps.pp_size > 1: + filename_parts.append(f"PP-{self.ps.pp_rank}") + if self.ps.moe_ep_size > 1: + filename_parts.append(f"EP-{self.ps.moe_ep_rank}") filename = ( (self.output_prefix + "-" if self.output_prefix else "")