Repository navigation
[None][infra] unify GB300 pre-merge and post-merge test lists - #21327
ZhanruiSunCh wants to merge 4 commits into
Conversation
|
Note Reviews pausedIt looks like this branch is under active development. To avoid overwhelming you with review comments due to an influx of new commits, CodeRabbit has automatically paused this review. You can configure this behavior by changing the
Use the following commands to manage reviews:
Use the checkboxes below for quick actions:
No actionable comments were generated in the recent review. WalkthroughThe GB300 single-GPU list now requires exactly one GPU and has an empty test list. The multi-GPU list adds three pre-merge groups. Jenkins selection, the related-file trigger, rule documentation, and a topology assertion are updated. ChangesGB300 CI test lists
Estimated code review effort: 2 (Simple) | ~15 minutes Change: Refactor Suggested reviewers: Merge Risk:
|
|
/bot run --stage-list="GB300-4_GPUs-PyTorch-1,GB300-4_GPUs-PyTorch-Post-Merge-1,GB300-4_GPUs-PyTorch-Post-Merge-2,GB300-4_GPUs-PyTorch-Post-Merge-3" |
|
PR_Github #81654 [ run ] triggered by Bot. Commit: |
|
PR_Github #81654 [ run ] completed with state
|
| - condition: | ||
| ranges: | ||
| system_gpu_count: | ||
| gte: 1 |
There was a problem hiding this comment.
This list only maps to a 4-GPU stage currently. If some tests only require 1-GPU, I think we should keep l0_gb300 and l0_gb300_multi_gpus as similar to others, while l0_gb300 only contains the single-GPU tests with a dedicated 1-GPU test stage.
There was a problem hiding this comment.
There is currently no dedicated GB300 1-GPU stage; the existing GB300 functional stages request 4 GPUs. In commit a18ecb3863, I retained l0_gb300.yml as an empty placeholder with the basic condition structure and a comment explaining that no dedicated test stage currently uses it. Existing tests remain in l0_gb300_multi_gpus.yml, preserving the current scheduling and test coverage.
a18ecb3 to
bb74f90
Compare
|
/bot run --stage-list="GB300-4_GPUs-PyTorch-1,GB300-4_GPUs-PyTorch-Post-Merge-1,GB300-4_GPUs-PyTorch-Post-Merge-2,GB300-4_GPUs-PyTorch-Post-Merge-3" |
|
PR_Github #82258 [ run ] triggered by Bot. Commit: |
|
PR_Github #82258 [ run ] completed with state
|
Move the post-merge GB300 blocks into l0_gb300.yml and point the existing three post-merge stages at the shared context. Preserve all conditions, ordered test entries, stage names, GPU allocations and split counts. Remove the obsolete list, update references and keep the multi-GPU review group assigned to the merged list. Signed-off-by: ZhanruiSunCh <184402041+ZhanruiSunCh@users.noreply.github.com>
Use l0_gb300_multi_gpus.yml for both pre-merge and post-merge coverage. Update the pre-merge stage and related references while preserving test conditions, selections, resources, and sharding. The existing multi-GPU CODEOWNERS pattern covers the consolidated list. Signed-off-by: ZhanruiSunCh <184402041+ZhanruiSunCh@users.noreply.github.com>
Keep the l0_gb300 YAML structure with an empty tests list and document that no dedicated GB300 single-GPU stage currently uses it. Existing coverage remains in the four-GPU stages. Signed-off-by: ZhanruiSunCh <184402041+ZhanruiSunCh@users.noreply.github.com>
Point the Kimi K3 topology regression assertion added on main at the consolidated multi-GPU list. Keep its expected GPU counts and stage mappings unchanged. Signed-off-by: ZhanruiSunCh <184402041+ZhanruiSunCh@users.noreply.github.com>
bb74f90 to
4cdd670
Compare
|
/bot run --disable-fail-fast --stage-list="GB300-4_GPUs-PyTorch-1,GB300-4_GPUs-PyTorch-Post-Merge-1,GB300-4_GPUs-PyTorch-Post-Merge-2,GB300-4_GPUs-PyTorch-Post-Merge-3" |
|
PR_Github #82397 [ run ] triggered by Bot. Commit: |
|
PR_Github #82397 [ run ] completed with state
|
Description
GB300 functional stages already allocate four GPUs. Consolidate their pre-merge and post-merge coverage into
l0_gb300_multi_gpus.yml, retaining the existing condition blocks and updating the pre-merge stage to use the shared context.Preserve test entries and ordering, stage names, GPU allocations, and shard counts. Retain
l0_gb300.ymlas an empty single-GPU placeholder with its basic condition structure and a comment stating that no dedicated test stage currently uses it. Remove its unused static trigger entry. The existing multi-GPU CODEOWNERS pattern covers the shared list. Update the Kimi K3 topology assertion to use the shared list name, keeping its expected GPU counts and stage mappings unchanged.Test Coverage
20bf00803a; all current test entries and conditions are preserved in the shared list.trt-test-db==1.8.5+bc6df7produced identical before/after outputs in 16 scenarios: GB110/GB300, pre/post-merge, and 1/2/4/8 GPUs. Four-GPU selections contain 92 pre-merge and 84 post-merge entries before sharding.0:53), without pytest results from the failing allocations. A read-only Jenkins diagnostic found both affected nodes subsequently drained for reported Lustre issues. GPU validation is being rerun on the rebased head with--disable-fail-fastso sibling stages can finish.PR Checklist
Dev Engineer Review
The change consolidates GB300 pre-merge test selections in
l0_gb300_multi_gpus.yml. The four-GPU pre-merge stage now uses that list.l0_gb300.ymlremains an empty single-GPU placeholder, and changes to it no longer trigger multi-GPU testing. The Kimi K3 topology assertion now uses the shared list name.The author reports that test selections matched across 16 scenarios, CBTS ordered selections matched across 123 parsed stages, and test-to-stage mappings were unchanged. The author also reports 25 modeling_v2 CBTS tests passed. The reported GPU pipelines failed during SLURM allocation and returned no pytest results for the affected allocations. Successful GPU validation on the rebased head is not reported.
QA Engineer Review
The test-list changes consolidate GB300 pre-merge entries for PyTorch, models, kernels, serving, Engram sharding, DeepSeek, NVFP4 DSpark, GLM-5.3-Flash, and Kimi K3 in
l0_gb300_multi_gpus.yml. The topology unit-test expectation now refers to that list. The author reports 25 modeling_v2 CBTS tests passed, but GPU validation remains unresolved because the reported pipelines failed during allocation without pytest results. Coverage verdict: needs follow-up.Per-File QA Perspective
jenkins/L0_MergeRequest.groovy: Removesl0_gb300.ymlfrom related-file trigger handling. Verify that changes to the shared list still trigger the intended multi-GPU tests.jenkins/L0_Test.groovy: Switches the GB300 four-GPU pre-merge stage tol0_gb300_multi_gpus. Verify selected tests and resource settings.jenkins/scripts/cbts/cbts/rules/README.md: Clarifies that thecomm/entry underl0_gb300_multi_gpusis post-merge. Check that this matches the list configuration.tests/integration/test_lists/test-db/l0_gb300.yml: Removes the former entries and retains an empty single-GPU placeholder. Verify CI selection behavior for its condition.tests/integration/test_lists/test-db/l0_gb300_multi_gpus.yml: Adds GB300 pre-merge groups for 1–4-GPU and four-GPU configurations. Verify selectors, ordering, and shard settings in CI.tests/unittest/scripts/cbts/test_modeling_v2_rule.py: Updates the expected Kimi K3 topology list name. The author reports 25 modeling_v2 CBTS tests passed.