[None][infra] Split stage GB200-4_GPUs-PyTorch-*

[EmmaQiaoCh]

Summary by CodeRabbit

• Tests
  • Enhanced test infrastructure configuration for GPU performance testing.
  • Expanded test execution stages to improve parallelization and enable more efficient distribution of workloads across multiple test execution groups.
  • Optimized test distribution parameters to provide better resource allocation during test runs and improve overall testing efficiency and reliability.

Description

Test Coverage

PR Checklist

Please review the following before submitting your PR:

• PR description clearly explains what and why. If using CodeRabbit's summary, please make sure it makes sense.

• PR Follows TRT-LLM CODING GUIDELINES to the best of your knowledge.

• Test cases are provided for new code paths (see test instructions)

• If PR introduces API changes, an appropriate PR label is added - either api-compatible or api-breaking. For api-breaking, include BREAKING in the PR title.

• Any new dependencies have been scanned for license and vulnerabilities

• CODEOWNERS updated if ownership changes

• Documentation updated as needed

• Update tava architecture diagram if there is a significant design change in PR.

• The reviewers assigned automatically/manually are appropriate for the PR.

• Please check this after reviewing the above items as appropriate for this PR.

GitHub Bot Help

To see a list of available CI bot commands, please comment /bot help.

[EmmaQiaoCh]

/bot run --stage-list "GB200-4_GPUs-PyTorch-1,GB200-4_GPUs-PyTorch-2,GB200-4_GPUs-PyTorch-3,GB200-4_GPUs-PyTorch-4"

[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: ee3ff027-46fb-4b83-859c-9318639fa659

📥 Commits

Reviewing files that changed from the base of the PR and between 8be182d and c400623.

📒 Files selected for processing (1)

• jenkins/L0_Test.groovy

---

📝 Walkthrough

Walkthrough

The PR updates Jenkins SBSA SLURM test stage configuration for GB200-4_GPUs-PyTorch, increasing test parallelization from 2 to 4 groups. Two new stage entries are added and the split-count parameter is updated to match the expanded group count.

Changes

SBSA GPU Test Parallelization Configuration

Layer / File(s)	Summary
SBSA GPU test stage fan-out expansion jenkins/L0_Test.groovy	SBSASlurmTestConfigs map for GB200-4_GPUs-PyTorch is updated to distribute test load across 4 stage groups instead of 2 by adding entries -3 and -4, and updating the split-count from 2 to 4 for entries -1 and -2.

🎯 1 (Trivial) | ⏱️ ~3 minutes

Possibly related PRs

• NVIDIA/TensorRT-LLM#14661: Updates Jenkins SLURM test stage split-counts and fan-out configuration for other GPU test stages in the same configuration map.

Suggested reviewers

• niukuo
• zeroepoch

🚥 Pre-merge checks | ✅ 4 | ❌ 1

❌ Failed checks (1 warning)

Check name	Status	Explanation	Resolution
Description check	⚠️ Warning	The PR description is empty except for the template structure; the Description and Test Coverage sections lack any substantive content explaining the change or test coverage.	Add a detailed description explaining why the stage is being split, the expected impact, and list relevant tests that verify the stage configuration changes.

✅ Passed checks (4 passed)

Check name	Status	Explanation
Title check	✅ Passed	The title clearly describes the main change: splitting the GB200-4_GPUs-PyTorch test stage into multiple stage entries, which matches the code modifications.
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.

_{✏️ Tip: You can configure your own custom pre-merge checks in the settings.}

✨ Finishing Touches

🧪 Generate unit tests (beta)

• Create PR with unit tests

---

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

[tensorrt-cicd]

PR_Github #52623 [ run ] triggered by Bot. Commit: c400623 Link to invocation

[EmmaQiaoCh]

Close this pr since Tyler already have one to fix this.

[tensorrt-cicd]

PR_Github #52623 [ run ] completed with state SUCCESS. Commit: c400623
/LLM/main/L0_MergeRequest_PR pipeline #41905 (Partly Tested) completed with status: 'SUCCESS'

CI Report

Link to invocation