From b9cf629bd0e924c69f3d8bfefbfdb77df5ffc7be Mon Sep 17 00:00:00 2001 From: Frederik Gossen Date: Mon, 20 Apr 2026 13:31:03 -0400 Subject: [PATCH] [Core] Label torch trace logging overhead with dynamo_timed (#39329) Signed-off-by: Frederik Gossen --- vllm/compilation/piecewise_backend.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/vllm/compilation/piecewise_backend.py b/vllm/compilation/piecewise_backend.py index 02a4dad5460..b647e0d8581 100644 --- a/vllm/compilation/piecewise_backend.py +++ b/vllm/compilation/piecewise_backend.py @@ -11,6 +11,7 @@ from typing import Any import torch._functorch.config import torch.fx as fx +from torch._dynamo.utils import dynamo_timed from torch._inductor.runtime.triton_heuristics import CachingAutotuner from torch._logging._internal import trace_structured @@ -275,6 +276,7 @@ class PiecewiseBackend: range_entry.compiled = True + @dynamo_timed("vllm_log_compile_start_torch_trace_only") def _log_compile_start(self, compile_range: Range): """Log compilation event for TORCH_TRACE/tlparse.""" is_cudagraph_size = (