[https://nvbugs/6428069][chore] Unwaive the pp=2 disagg tests - #18135
[https://nvbugs/6428069][chore] Unwaive the pp=2 disagg tests#18135lori-ren wants to merge 5 commits into
Conversation
NVIDIA#15920 revert Signed-off-by: Lori Ren <lorir@nvidia.com>
|
No actionable comments were generated in the recent review. 🎉 ℹ️ Recent review info⚙️ Run configurationConfiguration used: Path: .coderabbit.yaml Review profile: CHILL Plan: Enterprise Run ID: 📒 Files selected for processing (1)
💤 Files with no reviewable changes (1)
Included review availability: Your plan provides up to 12 included reviews per hour; 11 remain after this review. WalkthroughThe integration waiver list removes two Llama 3.1 8B symmetric skips and one TinyLlama disaggregated-serving skip. The Llama MMLU ChangesIntegration test waiver updates
Estimated code review effort: 1 (Trivial) | ~2 minutes Merge Risk: ⚪ Minimal · up to This change re-enables three previously waived pp=2 tests without altering product or runtime behavior; no actionable merge-blocking risk remains beyond normal checks and review. 🚥 Pre-merge checks | ✅ 5✅ Passed checks (5 passed)
Full details: Description checkExplanation The description explains the reason for the change, identifies the three tests being unwaived, documents CI coverage and results, and explains why the five pp=4 waivers remain. It includes the required Description, Test Coverage, and PR Checklist sections. Full details: Docstring CoverageExplanation No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check. Docstring coverage is scoped to functions touched by this diff. Analyzed 0 functions across 0 files. ✨ Finishing Touches🧪 Generate unit tests (beta)
Comment |
Signed-off-by: Lori Ren <lorir@nvidia.com> # Conflicts: # tests/integration/test_lists/waives.txt
|
/bot run --extra-stage "DGX_H200-8_GPUs-PyTorch-Post-Merge-1" |
|
PR_Github #68795 [ run ] triggered by Bot. Commit: |
|
PR_Github #68795 [ run ] completed with state
|
|
/bot run --extra-stage "DGX_H200-8_GPUs-PyTorch-Post-Merge-1" --disable-fail-fast |
|
PR_Github #68949 [ run ] triggered by Bot. Commit: |
|
PR_Github #68949 [ run ] completed with state
|
|
/bot run --extra-stage "DGX_H200-8_GPUs-PyTorch-Post-Merge-1" --disable-fail-fast |
|
PR_Github #69010 [ run ] triggered by Bot. Commit: |
|
PR_Github #69010 [ run ] completed with state
|
…nly the pp=2 cases CI confirmed Signed-off-by: Lori Ren <lorir@nvidia.com>
Signed-off-by: Lori Ren <lorir@nvidia.com> # Conflicts: # tests/integration/test_lists/waives.txt
|
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. |
|
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. |
Description
use_host_stop_criteria(added by #15920) was reverted wholesale by #16163, so the mechanismbehind nvbugs/6428069 no longer exists —
grep -rn use_host_stop_criteriareturns nothing onmain. The 8 waivers it left behind have been skipping these cases since 2026-07-08.Un-waiving all 8 was tried first. CI build #56392 ran them on
DGX_H200-8_GPUs-PyTorch-Post-Merge-1and split them exactly by pipeline-parallel degree:test_tp_pp_symmetric[GSM8K-tp2pp2],[MMLU-tp2pp2],test_disaggregated_ctxtp2pp2_gentp2pp2test_ctx_pp_gen_tp_asymmetric[…ctx_pp=4]×4,test_disaggregated_ctxpp4_genpp4This PR now un-waives only the 3 that CI confirmed passing. The 5
pp=4lines staywaived.
The
pp=4failures are a different defect from 6428069: a 300 s HangDetector fire with theNIXL KV transfer never completing.
IndexError,finish_reasonsanduse_host_stop_criteriaappear 0 times across all five logs. Per-rank stacks show 3 of 4 context ranks blocked in
a world-size allgather inside
_gen_consensus(transceiver.py:491) while the 4th is blockedin
wait_on_pp_send_handles(py_executor.py:3287) — a collective-participation deadlockthat a deeper pipeline makes reachable. Those 5 lines still cite 6428069 and should be
re-pointed at a bug for that hang.
Test Coverage
The three un-waived cases, on
DGX_H200-8_GPUs-PyTorch-Post-Merge-1:accuracy/test_disaggregated_serving.py::TestLlama3_1_8BInstruct::test_tp_pp_symmetric[GSM8K-tp2pp2]accuracy/test_disaggregated_serving.py::TestLlama3_1_8BInstruct::test_tp_pp_symmetric[MMLU-tp2pp2]disaggregated/test_disaggregated.py::test_disaggregated_ctxtp2pp2_gentp2pp2[TinyLlama-1.1B-Chat-v1.0]All three passed in build #56392. Re-run with
/bot run --extra-stage "DGX_H200-8_GPUs-PyTorch-Post-Merge-1"(needs theci: full pre-merge approvedlabel, or the multi-GPU stages are skipped).Note for reviewers
waives.txthas neighbouring entries under nvbugs/6655360, 6644475 and 6611817 coveringother variants of these same test functions. Those are a different defect again — a
disaggregated-server hang/timeout at launch on H100 4-GPU — and are intentionally left alone.
PR Checklist
GitHub Bot Help
To see a list of available CI bot commands, please comment
/bot help.Dev Engineer Review
tests/integration/test_lists/waives.txt.QA Engineer Review
tests/integration/test_lists/waives.txt.test-db/orqa/files were modified.use_host_stop_criteriawas removed.Verdict: needs follow-up