[https://nvbugs/6428087][fix] Extend the PP send payload with the flag (backward-compatible 2/3-tuple…

[trtllm-agent]

Summary

• Root cause: PP send/recv drops SampleStateTorch.use_host_stop_criteria, so non-last PP ranks default to False and call process_draft_tokens with an empty finish_reasons list from the last rank's host fast path.
• Fix: Extend the PP send payload with the flag (backward-compatible 2/3-tuple handling) so all PP ranks pick the same update branch; remove the waiver.
• 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/6428087

Dev Engineer Review

• Propagates SampleStateTorch.use_host_stop_criteria across pipeline-parallel ranks.
• Supports both legacy 2-tuple payloads and new 3-tuple payloads.
• Removes the waiver for NVBug 6428087 and updates the related waiver reference to NVBug 6427411.
• The changes are consistent with the reported bug fix. No public API changes are introduced.

QA Engineer Review

• Only tests/integration/test_lists/waives.txt changed.
• Removed the waived TestDeepSeekV3Lite::test_nvfp4_4gpus CUTEDSL/tp2pp2 configuration linked to NVBug 6428087.
• Added the revised configuration linked to NVBug 6427411.
• No test-db/ or qa/ files were modified.
• Verdict: needs follow-up because CBTS coverage data is unavailable.

[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: 4a1bd0c0-f6bb-49a1-9a27-164e0a57c707

📥 Commits

Reviewing files that changed from the base of the PR and between 1cef02e and c180e55.

📒 Files selected for processing (2)

• tensorrt_llm/_torch/pyexecutor/py_executor.py
• tests/integration/test_lists/waives.txt

💤 Files with no reviewable changes (1)

• tests/integration/test_lists/waives.txt

🚧 Files skipped from review as they are similar to previous changes (1)

• tensorrt_llm/_torch/pyexecutor/py_executor.py

---

Walkthrough

The PR extends the SAMPLE_STATE PP ring broadcast to propagate use_host_stop_criteria alongside existing state data. It also replaces a DeepSeekV3Lite test waiver configuration and NVBUG reference.

Changes

PP ring broadcast and test waiver

Layer / File(s)	Summary
Propagate use_host_stop_criteria in PP ring broadcast tensorrt_llm/_torch/pyexecutor/py_executor.py	The send path adds use_host_stop_criteria to the SAMPLE_STATE payload. The receive path applies the value to sample_state when present.
Update DeepSeekV3Lite waiver tests/integration/test_lists/waives.txt	The waiver uses a different CUTEDSL/tp2pp2 configuration and changes the linked NVBUG from 6428087 to 6427411.

Estimated code review effort: 2 (Simple) | ~10 minutes

Possibly related PRs

• NVIDIA/TensorRT-LLM#17375: Updates a related DeepSeekV3Lite tp2pp2 waiver with a different configuration and NVBUG.
• NVIDIA/TensorRT-LLM#17398: Modifies another TestDeepSeekV3Lite waiver entry.
• NVIDIA/TensorRT-LLM#17424: Updates test waivers in the same integration waiver file.

Suggested reviewers: qijune, bowenfu, yingguo-trt

🚥 Pre-merge checks | ✅ 5

✅ Passed checks (5 passed)

Check name	Status	Explanation
Title check	✅ Passed	The title clearly identifies the bug fix and the PP payload change.
Description check	✅ Passed	The description explains the root cause and fix, lists test coverage, and links the NVBug; the template checklist is not included but is non-critical.
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.}

[trtllm-agent]

NVBug 6428087 is closed as Bug - Fixed. The linked bug appears resolved elsewhere or for a reason that does not prove this PR is redundant. This PR should be judged on its own merits; repair-bot is not auto-closing it.

[coderabbitai]

Note

GitHub couldn't provide a complete incremental comparison for this pull request, so CodeRabbit is performing a full review instead. This review may take a little longer.