From b970e993cff71257f86b136f0653b2706374b95f Mon Sep 17 00:00:00 2001 From: Chenfei Zhang Date: Wed, 22 Jul 2026 03:46:11 -0700 Subject: [PATCH] [nvbugs/6484343][fix] Rename MTPEagleDynamicTreeWorker.forward to _forward_impl Signed-off-by: Chenfei Zhang --- tensorrt_llm/_torch/speculative/mtp_dynamic_tree.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorrt_llm/_torch/speculative/mtp_dynamic_tree.py b/tensorrt_llm/_torch/speculative/mtp_dynamic_tree.py index 5dca84abae8e..053d23362f98 100644 --- a/tensorrt_llm/_torch/speculative/mtp_dynamic_tree.py +++ b/tensorrt_llm/_torch/speculative/mtp_dynamic_tree.py @@ -636,7 +636,7 @@ def _relocate_kv_eagerly(self, attn_metadata, batch_size): # Top-level forward # # ------------------------------------------------------------------ # @nvtx_range("mtp_dyn.forward") - def forward( + def _forward_impl( self, input_ids, position_ids,