From bcf66ef3e1f11340e76725166daf854a4edd2077 Mon Sep 17 00:00:00 2001
From: fzyzcjy <5236035+fzyzcjy@users.noreply.github.com>
Date: Thu, 5 Jun 2025 15:07:03 +0800
Subject: [PATCH] Tiny allow profiler API to auto create directory (#6865)

---
 python/sglang/srt/managers/scheduler.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py
index 5c2141d77..52271e563 100644
--- a/python/sglang/srt/managers/scheduler.py
+++ b/python/sglang/srt/managers/scheduler.py
@@ -24,6 +24,7 @@ from collections import defaultdict, deque
 from concurrent import futures
 from dataclasses import dataclass
 from http import HTTPStatus
+from pathlib import Path
 from types import SimpleNamespace
 from typing import Dict, List, Optional, Tuple, Union
 
@@ -2283,6 +2284,9 @@ class Scheduler(
                 message="Profiling is not in progress. Call /start_profile first.",
             )
 
+        if not Path(self.torch_profiler_output_dir).exists():
+            Path(self.torch_profiler_output_dir).mkdir(parents=True, exist_ok=True)
+
         stage_suffix = f"-{stage.__str__()}" if stage else ""
         logger.info("Stop profiling" + stage_suffix + "...")
         if self.torch_profiler is not None: