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 = (