[https://nvbugs/6484343][fix] Renamed MTPEagleDynamicTreeWorker.forward → _forward_impl; the base…

[chenfeiz0326]

Summary

• Root cause: MTPEagleDynamicTreeWorker defined a forward method, which SpecWorkerBase.init_subclass (guard from nvbugs/6442074) forbids, raising TypeError at import and collecting 0 tests.
• Fix: Renamed MTPEagleDynamicTreeWorker.forward → _forward_impl; the base wrapper now drives it. No internal callers of self.forward exist, so the rename is safe.
• 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/6484343

Dev Engineer Review

• Renamed MTPEagleDynamicTreeWorker.forward to _forward_impl to satisfy the SpecWorkerBase.__init_subclass__ guard.
• Preserved the method signature, implementation, and NVTX label (mtp_dyn.forward), avoiding behavioral or performance changes.
• The base wrapper now targets the renamed implementation, with no remaining internal self.forward callers.
• No configuration or test-list files were changed.
• Verdict: sufficient.

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: feadf9ca-1636-42a5-a61f-d321307f2973

📥 Commits

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

📒 Files selected for processing (1)

• tensorrt_llm/_torch/speculative/mtp_dynamic_tree.py

---

Walkthrough

The MTP dynamic tree worker method is renamed from forward to _forward_impl, with its parameters, implementation, and NVTX annotation unchanged.

Changes

MTP dynamic tree entry method

Layer / File(s)	Summary
Rename worker entry method tensorrt_llm/_torch/speculative/mtp_dynamic_tree.py	MTPEagleDynamicTreeWorker.forward is renamed to _forward_impl without changing the method body or signature parameters.

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

Possibly related PRs

• NVIDIA/TensorRT-LLM#16724: Makes the same entry-method rename in the same file.
• NVIDIA/TensorRT-LLM#16727: Makes the same entry-method rename while preserving the NVTX decoration.
• NVIDIA/TensorRT-LLM#16728: Makes the same rename to address the forbidden base forward override.

Suggested reviewers: sunnyqgg

🚥 Pre-merge checks | ✅ 5

✅ Passed checks (5 passed)

Check name	Status	Explanation
Title check	✅ Passed	The title is specific and matches the main change: renaming the worker method to fix the import-time guard issue.
Description check	✅ Passed	The description covers the root cause, fix, and test plan, but it does not follow the template headings exactly.
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.}