From 11943df41202e6135deb867660f658127f0cecbd Mon Sep 17 00:00:00 2001 From: Chenfei Zhang Date: Wed, 22 Jul 2026 03:35:03 -0700 Subject: [PATCH] [nvbugs/6478697][fix] Rename MTPEagleDynamicTreeWorker.forward to _forward_impl MTPEagleDynamicTreeWorker (added by #15582) overrode SpecWorkerBase.forward, which the __init_subclass__ guard (nvbugs/6442074) forbids. This raised a TypeError at import time, breaking all pytest collection. Rename the method to _forward_impl, matching the MTPWorker pattern, so the base forward wrapper provides the guaranteed spec-dec attn-metadata cleanup. 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,