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 "")