Skip to content

Conversation

@dongfengy
Copy link
Collaborator

@dongfengy dongfengy commented Sep 16, 2025

Summary by CodeRabbit

  • Documentation
    • Updated the quick-start deployment guide with revised configuration examples.
    • Increased default stream interval values in configuration snippets.
    • Added guidance on attention data-parallel settings, including balance enablement, batching wait iterations, and timeout iterations.

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)

  • Any new dependencies have been scanned for license and vulnerabilities

  • CODEOWNERS updated if ownership changes

  • Documentation updated as needed

  • 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

/bot [-h] ['run', 'kill', 'skip', 'reuse-pipeline'] ...

Provide a user friendly way for developers to interact with a Jenkins server.

Run /bot [-h|--help] to print this help message.

See details below for each supported subcommand.

Details

run [--reuse-test (optional)pipeline-id --disable-fail-fast --skip-test --stage-list "A10-PyTorch-1, xxx" --gpu-type "A30, H100_PCIe" --test-backend "pytorch, cpp" --add-multi-gpu-test --only-multi-gpu-test --disable-multi-gpu-test --post-merge --extra-stage "H100_PCIe-TensorRT-Post-Merge-1, xxx" --detailed-log --debug(experimental)]

Launch build/test pipelines. All previously running jobs will be killed.

--reuse-test (optional)pipeline-id (OPTIONAL) : Allow the new pipeline to reuse build artifacts and skip successful test stages from a specified pipeline or the last pipeline if no pipeline-id is indicated. If the Git commit ID has changed, this option will be always ignored. The DEFAULT behavior of the bot is to reuse build artifacts and successful test results from the last pipeline.

--disable-reuse-test (OPTIONAL) : Explicitly prevent the pipeline from reusing build artifacts and skipping successful test stages from a previous pipeline. Ensure that all builds and tests are run regardless of previous successes.

--disable-fail-fast (OPTIONAL) : Disable fail fast on build/tests/infra failures.

--skip-test (OPTIONAL) : Skip all test stages, but still run build stages, package stages and sanity check stages. Note: Does NOT update GitHub check status.

--stage-list "A10-PyTorch-1, xxx" (OPTIONAL) : Only run the specified test stages. Examples: "A10-PyTorch-1, xxx". Note: Does NOT update GitHub check status.

--gpu-type "A30, H100_PCIe" (OPTIONAL) : Only run the test stages on the specified GPU types. Examples: "A30, H100_PCIe". Note: Does NOT update GitHub check status.

--test-backend "pytorch, cpp" (OPTIONAL) : Skip test stages which don't match the specified backends. Only support [pytorch, cpp, tensorrt, triton]. Examples: "pytorch, cpp" (does not run test stages with tensorrt or triton backend). Note: Does NOT update GitHub pipeline status.

--only-multi-gpu-test (OPTIONAL) : Only run the multi-GPU tests. Note: Does NOT update GitHub check status.

--disable-multi-gpu-test (OPTIONAL) : Disable the multi-GPU tests. Note: Does NOT update GitHub check status.

--add-multi-gpu-test (OPTIONAL) : Force run the multi-GPU tests in addition to running L0 pre-merge pipeline.

--post-merge (OPTIONAL) : Run the L0 post-merge pipeline instead of the ordinary L0 pre-merge pipeline.

--extra-stage "H100_PCIe-TensorRT-Post-Merge-1, xxx" (OPTIONAL) : Run the ordinary L0 pre-merge pipeline and specified test stages. Examples: --extra-stage "H100_PCIe-TensorRT-Post-Merge-1, xxx".

--detailed-log (OPTIONAL) : Enable flushing out all logs to the Jenkins console. This will significantly increase the log volume and may slow down the job.

--debug (OPTIONAL) : Experimental feature. Enable access to the CI container for debugging purpose. Note: Specify exactly one stage in the stage-list parameter to access the appropriate container environment. Note: Does NOT update GitHub check status.

For guidance on mapping tests to stage names, see docs/source/reference/ci-overview.md
and the scripts/test_to_stage_mapping.py helper.

kill

kill

Kill all running builds associated with pull request.

skip

skip --comment COMMENT

Skip testing for latest commit on pull request. --comment "Reason for skipping build/test" is required. IMPORTANT NOTE: This is dangerous since lack of user care and validation can cause top of tree to break.

reuse-pipeline

reuse-pipeline

Reuse a previous pipeline to validate current commit. This action will also kill all currently running builds associated with the pull request. IMPORTANT NOTE: This is dangerous since lack of user care and validation can cause top of tree to break.

Signed-off-by: Dongfeng Yu <dongfengy@nvidia.com>
@dongfengy dongfengy requested a review from a team as a code owner September 16, 2025 22:31
@coderabbitai
Copy link
Contributor

coderabbitai bot commented Sep 16, 2025

📝 Walkthrough

Walkthrough

Documentation update to the TRTLLM quick-start guide: increased stream_interval values in TRTLLM and CUTLASS config examples, and added attention_dp_config with balance and batching parameters in the CUTLASS block.

Changes

Cohort / File(s) Summary
Docs: TRTLLM Quick Start config updates
docs/source/deployment-guide/quick-start-recipe-for-gpt-oss-on-trtllm.md
- TRTLLM config: stream_interval 10 → 20
- CUTLASS config: stream_interval 10 → 20
- CUTLASS: added attention_dp_config { enable_balance: true, batching_wait_iters: 50, timeout_iters: 1 }

Estimated code review effort

🎯 2 (Simple) | ⏱️ ~8 minutes

✨ Finishing touches
🧪 Generate unit tests
  • Create PR with unit tests
  • Post copyable unit tests in a comment

Tip

👮 Agentic pre-merge checks are now available in preview!

Pro plan users can now enable pre-merge checks in their settings to enforce checklists before merging PRs.

  • Built-in checks – Quickly apply ready-made checks to enforce title conventions, require pull request descriptions that follow templates, validate linked issues for compliance, and more.
  • Custom agentic checks – Define your own rules using CodeRabbit’s advanced agentic capabilities to enforce organization-specific policies and workflows. For example, you can instruct CodeRabbit’s agent to verify that API documentation is updated whenever API schema files are modified in a PR. Note: Upto 5 custom checks are currently allowed during the preview period. Pricing for this feature will be announced in a few weeks.

Please see the documentation for more information.

Example:

reviews:
  pre_merge_checks:
    custom_checks:
      - name: "Undocumented Breaking Changes"
        mode: "warning"
        instructions: |
          Pass/fail criteria: All breaking changes to public APIs, CLI flags, environment variables, configuration keys, database schemas, or HTTP/GraphQL endpoints must be documented in the "Breaking Change" section of the PR description and in CHANGELOG.md. Exclude purely internal or private changes (e.g., code not exported from package entry points or explicitly marked as internal).

Please share your feedback with us on this Discord post.


Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

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

Pre-merge checks

❌ Failed checks (1 warning)
Check name Status Explanation Resolution
Description Check ⚠️ Warning The PR body currently contains the repository PR template text rather than a completed description and is missing required information. It does not include a properly formatted PR title per the template, a concrete "Description" explaining what was changed and why (the PR objectives indicate this is a cherry-pick of a deployment-guide update), nor a "Test Coverage" section listing relevant tests or CI expectations. The raw_summary shows a docs change to docs/source/deployment-guide/quick-start-recipe-for-gpt-oss-on-trtllm.md (stream_interval adjustments and an attention_dp_config addition), but those concrete changes and the cherry-picked source/commit are not documented in the PR body. Because the required template sections are empty or incomplete, the description check fails. Please complete the PR description to follow the repository template: set a title using the required format (e.g., [None][doc] Cherry-pick deployment guide update from 1.1.0rc2), provide a concise Description that lists the files changed, the exact commits or original PR number(s) being cherry-picked, and the rationale for the cherry-pick, and add a Test Coverage section specifying any tests or CI stages to run (or note "docs-only" if applicable). Also update the PR checklist (CODEOWNERS if ownership changes), assign appropriate reviewers, and indicate whether CI should be re-run; once updated, request reviewers or re-run the pipeline so the change can be reviewed.
✅ Passed checks (2 passed)
Check name Status Explanation
Title Check ✅ Passed The PR title "[None][doc] Cherry-pick deployment guide update from 1.1.0rc2 branch to main branch" follows the repository template and accurately identifies the primary change as a documentation cherry-pick from 1.1.0rc2 into main, which matches the changeset (deployment guide updates). It is clear and directly related to the main change, though slightly verbose.
Docstring Coverage ✅ Passed No functions found in the changes. Docstring coverage check skipped.

Copy link
Contributor

@coderabbitai coderabbitai bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 0

🧹 Nitpick comments (6)
docs/source/deployment-guide/quick-start-recipe-for-gpt-oss-on-trtllm.md (6)

68-78: Use a quoted heredoc to avoid accidental shell expansion; keep YAML indentation consistent.

  • Quote EOF to prevent variable expansion in copied configs.
  • Use consistent 2‑space indent under moe_config (matches cuda_graph_config block).

Apply this diff:

-EXTRA_LLM_API_FILE=/tmp/config.yml
-
-cat << EOF > ${EXTRA_LLM_API_FILE}
+EXTRA_LLM_API_FILE=/tmp/config.yml
+
+cat <<'EOF' > "${EXTRA_LLM_API_FILE}"
 enable_attention_dp: false
 cuda_graph_config:
   enable_padding: true
   max_batch_size: 720
 moe_config:
-    backend: TRTLLM
+  backend: TRTLLM
 stream_interval: 20
 num_postprocess_workers: 4
 EOF

75-75: Document the rationale/trade‑off for stream_interval: 20.

Increasing stream_interval reduces host I/O but can increase perceived latency for streamed tokens. Add a brief note on when to raise/lower it (e.g., throughput vs. TTFT/ITL targets).


85-99: Mirror heredoc quoting and indentation; minor YAML hygiene.

Same quoting/indent consistency as the TRTLLM block; keeps examples copy‑safe and uniform.

Apply this diff:

-EXTRA_LLM_API_FILE=/tmp/config.yml
-
-cat << EOF > ${EXTRA_LLM_API_FILE}
+EXTRA_LLM_API_FILE=/tmp/config.yml
+
+cat <<'EOF' > "${EXTRA_LLM_API_FILE}"
 enable_attention_dp: true
 cuda_graph_config:
   enable_padding: true
   max_batch_size: 720
 moe_config:
-    backend: CUTLASS
+  backend: CUTLASS
 stream_interval: 20
 num_postprocess_workers: 4
 attention_dp_config:
-    enable_balance: true
-    batching_wait_iters: 50
-    timeout_iters: 1
+  enable_balance: true
+  batching_wait_iters: 50
+  timeout_iters: 1
 EOF

222-223: Fix typo and remove lingering TODO before merge.

“Compeletions” → “Completions”; replace the placeholder text or drop the entire TODO.

Apply this diff:

-TODO: Use Chat Compeletions API / Responses API as the example after the PR is merged.
+Note: Example below will be updated to use the Chat Completions / Responses API after the related PR merges. (If already merged, replace this line with the final example.)

25-31: Minor wording polish: “support matrix” → “support matrices”; clarify default.

Consider: “There are multiple MoE backends in TRT‑LLM. The table below summarizes supported configurations. The default backend is CUTLASS.”


55-57: Add stability note about rc vs non‑rc images.

Since rc images are weekly and may change behavior, add a one‑liner reminding users to pin exact tags for reproducibility.

📜 Review details

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Pro

📥 Commits

Reviewing files that changed from the base of the PR and between a49cfb3 and 990c99e.

📒 Files selected for processing (1)
  • docs/source/deployment-guide/quick-start-recipe-for-gpt-oss-on-trtllm.md (2 hunks)
⏰ Context from checks skipped due to timeout of 90000ms. You can increase the timeout in your CodeRabbit configuration to a maximum of 15 minutes (900000ms). (1)
  • GitHub Check: Pre-commit Check
🔇 Additional comments (3)
docs/source/deployment-guide/quick-start-recipe-for-gpt-oss-on-trtllm.md (3)

46-47: Align container tag with the documented features/branch.

Example uses release:1.0.0rc6 while this PR cherry-picks from 1.1.0rc2. Please confirm the example image contains the attention DP options and stream_interval behavior shown below, or update the tag/note the minimum required version.


106-117: Good: server launch command is consistent with the YAML examples.

Flags align with the config blocks and Hugging Face model usage. No blockers.

If there’s a known minimum driver/runtime for Blackwell with attention DP enabled, consider adding it near Prerequisites.


92-97: Confirmed: option names match TorchLlmArgs — verify image tag

attention_dp_config.{enable_balance, batching_wait_iters, timeout_iters} and stream_interval are defined in tensorrt_llm/llmapi/llm_args.py; runtime/CLI variants (attention_dp_time_out_iters, attention_dp_batching_wait_iters, attention_dp_enable_balance) appear in tensorrt_llm/_torch/pyexecutor/config.py and examples/llm-api/quickstart_advanced.py. Confirm the serve image tag you use contains these definitions.

@dongfengy
Copy link
Collaborator Author

/bot run --disable-fail-fast

@tensorrt-cicd
Copy link
Collaborator

PR_Github #18834 [ run ] triggered by Bot

@tensorrt-cicd
Copy link
Collaborator

PR_Github #18834 [ run ] completed with state SUCCESS
/LLM/main/L0_MergeRequest_PR pipeline #14118 completed with status: 'SUCCESS'

@nv-guomingz nv-guomingz added the Cherry-pick It's a label that applies to Cherry-pick PR. label Sep 17, 2025
@dongfengy
Copy link
Collaborator Author

@nv-guomingz could you help merge? CI passed

@nv-guomingz nv-guomingz merged commit 026f22e into NVIDIA:main Sep 18, 2025
7 of 9 checks passed
dominicshanshan pushed a commit to dominicshanshan/TensorRT-LLM that referenced this pull request Sep 19, 2025
…to main branch (NVIDIA#7774)

Signed-off-by: Dongfeng Yu <dongfengy@nvidia.com>
Wong4j pushed a commit to Wong4j/TensorRT-LLM that referenced this pull request Sep 20, 2025
…to main branch (NVIDIA#7774)

Signed-off-by: Dongfeng Yu <dongfengy@nvidia.com>
MrGeva pushed a commit to nv-auto-deploy/TensorRT-LLM that referenced this pull request Sep 21, 2025
…to main branch (NVIDIA#7774)

Signed-off-by: Dongfeng Yu <dongfengy@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Cherry-pick It's a label that applies to Cherry-pick PR.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants