minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[Profiler] expand '~' for torch_profiler_output_dir (#11999)

Browse Source
This commit is contained in:
Muqi Li
2025-10-24 17:20:46 +08:00
committed by GitHub
parent b04cd3d487
commit 93ef9a094d
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/srt/managers/scheduler_profiler_mixin.py
+3 -4
View File
@@ -28,7 +28,7 @@ logger = logging.getLogger(__name__)
class SchedulerProfilerMixin: class SchedulerProfilerMixin:
def init_profiler(self): def init_profiler(self):
self.torch_profiler = None self.torch_profiler = None
self.torch_profiler_output_dir: Optional[str] = None self.torch_profiler_output_dir: Optional[Path] = None
self.profiler_activities: Optional[List[str]] = None self.profiler_activities: Optional[List[str]] = None
self.profile_id: Optional[str] = None self.profile_id: Optional[str] = None
self.profiler_start_forward_ct: Optional[int] = None self.profiler_start_forward_ct: Optional[int] = None
@@ -69,7 +69,7 @@ class SchedulerProfilerMixin:
if activities is None: if activities is None:
activities = ["CPU", "GPU"] activities = ["CPU", "GPU"]
self.torch_profiler_output_dir = output_dir self.torch_profiler_output_dir = Path(output_dir).expanduser()
self.torch_profiler_with_stack = with_stack self.torch_profiler_with_stack = with_stack
self.torch_profiler_record_shapes = record_shapes self.torch_profiler_record_shapes = record_shapes
self.profiler_activities = activities self.profiler_activities = activities
@@ -213,8 +213,7 @@ class SchedulerProfilerMixin:
message="Profiling is not in progress. Call /start_profile first.", message="Profiling is not in progress. Call /start_profile first.",
) )
if not Path(self.torch_profiler_output_dir).exists(): self.torch_profiler_output_dir.mkdir(parents=True, exist_ok=True)
Path(self.torch_profiler_output_dir).mkdir(parents=True, exist_ok=True)
stage_suffix = f"-{stage.name}" if stage else "" stage_suffix = f"-{stage.name}" if stage else ""
logger.info("Stop profiling" + stage_suffix + "...") logger.info("Stop profiling" + stage_suffix + "...")