From 132f1218e4937a451c49d4baa2759cdf94e9f820 Mon Sep 17 00:00:00 2001 From: qgai Date: Wed, 22 Jul 2026 02:47:04 -0700 Subject: [PATCH] [https://nvbugs/6442074][fix] Rename MTPEagleDynamicTreeWorker.forward to _forward_impl Signed-off-by: qgai --- 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,