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

Add Deepseek models into nightly tests (#12865)

Browse Source
This commit is contained in:
Kangyan-Zhou
2025-11-08 21:17:49 -08:00
committed by GitHub
parent f290e8016e
commit d134096319
18 changed files with 854 additions and 473 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/sglang/srt/managers/scheduler_profiler_mixin.py
View File

@@ -139,7 +139,7 @@ class SchedulerProfilerMixin:
schema.writeSchema(connection)
connection.commit()
del connection
torch.distributed.barrier(self.tp_cpu_group)
torch.distributed.barrier(self.cpu_group)
self.rpd_profiler = rpdTracerControl()
self.rpd_profiler.setPythonTrace(True)
@@ -236,14 +236,14 @@ class SchedulerProfilerMixin:
self.torch_profiler.export_chrome_trace(
os.path.join(self.torch_profiler_output_dir, filename)
)
torch.distributed.barrier(self.tp_cpu_group)
torch.distributed.barrier(self.cpu_group)
if self.rpd_profiler is not None:
self.rpd_profiler.rangePop()
self.rpd_profiler.stop()
self.rpd_profiler.flush()
torch.distributed.barrier(self.tp_cpu_group)
torch.distributed.barrier(self.cpu_group)
if self.tp_rank == 0:
from sglang.srt.utils.rpd_utils import rpd_to_chrome_trace