[Profiler] Add Nsight Systems support for serving (#1098)

Signed-off-by: Jinheng Li <ahengljh@gmail.com>
Signed-off-by: Canlin Guo <961750412@qq.com>
Co-authored-by: Claude Opus 4.5 <noreply@anthropic.com>
Co-authored-by: Canlin Guo <961750412@qq.com>
This commit is contained in:
authored and GitHub committed 2026-04-10 18:13:12 +08:00
1 parent fbb5dd5794
commit 78bef62f82
4 files changed
+276 -184

No files matched your search

+2 -6
View File
@@ -361,15 +361,11 @@ class DiffusionEngine:
)
def profile(self, is_start: bool = True, profile_prefix: str | None = None) -> None:
"""Start or stop torch profiling on all diffusion workers.
"""Start or stop profiling on all diffusion workers.
Args:
is_start: True to start profiling, False to stop.
profile_prefix: Optional prefix for trace filename (vLLM compat).
Note:
Matches vLLM's worker.profile() signature for consistency.
Traces are saved automatically via on_trace_ready callback.
profile_prefix: Optional prefix for trace filename.
"""
if is_start:
if profile_prefix is None: