[https://nvbugs/6478697][fix] Renamed mtp_dynamic_tree.py:639 forward → _forward_impl (matching…

[chenfeiz0326]

Summary

• Root cause: MTPEagleDynamicTreeWorker (added by [None][feat] Support Nemotron dynamic-tree MTP decoding #15582) overrode SpecWorkerBase.forward, which the init_subclass guard (nvbugs/6442074) forbids, raising TypeError at import and breaking all pytest collection (0 items).
• Fix: Renamed mtp_dynamic_tree.py:639 forward → _forward_impl (matching MTPWorker), letting the base SpecWorkerBase.forward wrapper provide the cleanup finally block; staged ONLY that one file to avoid the LFS-blob corruption that rejected both prior attempts.
• Automated fix generated by repair-bot

Test plan

• Verify fix on the same GPU type as the original failure
• Check for regressions in related tests

Links

• Bug: https://nvbugs/6478697

Dev Engineer Review

• Renamed MTPEagleDynamicTreeWorker.forward to _forward_impl to comply with the SpecWorkerBase.__init_subclass__ guard.
• Preserved the existing implementation, signature, NVTX decoration, and dynamic-tree behavior.
• Allows the inherited SpecWorkerBase.forward wrapper to handle spec-dec attention-metadata cleanup and restores pytest collection.
• No configuration, test-list, or test-code changes.

QA Engineer Review

No test changes.

[coderabbitai]

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: bcdcead3-5f7f-4726-9458-7c09784d9609

📥 Commits

Reviewing files that changed from the base of the PR and between 9cd6a49 and 11943df.

📒 Files selected for processing (1)

• tensorrt_llm/_torch/speculative/mtp_dynamic_tree.py

---

Walkthrough

The dynamic-tree worker’s top-level method is renamed from forward to _forward_impl. Its implementation, call signature structure, NVTX decoration, and speculative decoding behavior remain unchanged.

Changes

Dynamic tree entry point

Layer / File(s)	Summary
Rename dynamic-tree forward entry point tensorrt_llm/_torch/speculative/mtp_dynamic_tree.py	MTPEagleDynamicTreeWorker.forward(...) is renamed to _forward_impl(...) without changing the method body.

Estimated code review effort: 1 (Trivial) | ~2 minutes

Possibly related PRs

• NVIDIA/TensorRT-LLM#16724: Applies the same MTPEagleDynamicTreeWorker.forward(...) to _forward_impl(...) rename.

Suggested reviewers: sunnyqgg

🚥 Pre-merge checks | ✅ 5

✅ Passed checks (5 passed)

Check name	Status	Explanation
Title check	✅ Passed	The title is specific, matches the code change, and identifies the bug ticket and fix.
Description check	✅ Passed	It explains the root cause, fix, and test plan, with only template-specific headings and checklist items missing.
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.

✨ Finishing Touches

🧪 Generate unit tests (beta)

• Create PR with unit tests

---

_{Comment @coderabbitai help to get the list of available commands.}

[yuxianq]

Duplicated with #16724