Decode input_audio media containers with PyAV & Update memory profiler (#31832)

Signed-off-by: Shiyan Deng <dsy842974287@meta.com>
Signed-off-by: Lianmin Zheng <lianminzheng@gmail.com>
Co-authored-by: Lianmin Zheng <lianminzheng@gmail.com>
Co-authored-by: Lucia Fang <116399278+luccafong@users.noreply.github.com>
This commit is contained in:
Shiyan Deng
2026-07-24 16:18:45 -07:00
committed by GitHub
co-authored by Lianmin Zheng Lucia Fang
parent 14d6e1d3b1
commit 962c076934
8 changed files with 442 additions and 63 deletions
@@ -251,7 +251,9 @@ class SchedulerProfilerManager:
self.profile_in_progress = True
if "MEM" in activities:
torch.cuda.memory._record_memory_history(max_entries=100000)
torch.cuda.memory._record_memory_history(
max_entries=envs.SGLANG_MEM_PROFILE_MAX_ENTRIES.get()
)
self.profile_in_progress = True
if "CUDA_PROFILER" in activities:
@@ -356,7 +358,8 @@ class SchedulerProfilerManager:
if self.profiler_activities is not None and "MEM" in self.profiler_activities:
memory_profile_path = os.path.join(
self.torch_profiler_output_dir,
str(time.time())
stage_prefix
+ str(time.time())
+ f"-TP-{self.ps.tp_rank}-memory"
+ stage_suffix
+ ".pickle",