Skip to content

[TRTLLM-12154][test] Add Qwen3-32B FP8 disagg stress test#14278

Open
brnguyen2 wants to merge 4 commits into
NVIDIA:mainfrom
brnguyen2:brnguyen/qwen3-32b-fp8-disagg-stress
Open

[TRTLLM-12154][test] Add Qwen3-32B FP8 disagg stress test#14278
brnguyen2 wants to merge 4 commits into
NVIDIA:mainfrom
brnguyen2:brnguyen/qwen3-32b-fp8-disagg-stress

Conversation

@brnguyen2
Copy link
Copy Markdown
Collaborator

@brnguyen2 brnguyen2 commented May 18, 2026

Summary

Adds Qwen3-32B FP8 disaggregated serving coverage for the Eagle3 path on 8 GPUs.

The new disagg config uses:

  • target model: Qwen3/Qwen3-32B-FP8
  • Eagle3 draft model: Zhi-Create-Qwen3-32B-Eagle3, resolved relative to llm_models_root()
  • 4 context workers with TP1 and 1 generation worker with TP4
  • FP8 KV cache, chunked prefill, block/partial reuse, and DEFAULT cache transceiver
  • top-level speculative_config so context and generation workers agree on the cache-state handshake

The disagg harness now resolves relative model and speculative_config.speculative_model values through llm_models_root() while preserving absolute paths. This keeps the YAML portable across CI/local model roots.

Also makes the GSM8K lm-eval parser tolerate padded table cells, matching the output shape emitted by the local-completions run.

Test entries:

  • test_disaggregated_qwen3_32b_fp8 smoke test in l0_dgx_h200.yml
  • test_disaggregated_stress_test::qwen3_32b_fp8_stress in qa/llm_function_stress.txt, request_count=10000, accuracy_threshold=0.42

Refs: TRTLLM-12154

Test plan

  • Targeted smoke test passed before the upstream rebase with FP8 KV + Eagle3
  • Rebuild on upstream main after rebase
  • Rerun tests/integration/defs/disaggregated/test_disaggregated.py::test_disaggregated_qwen3_32b_fp8 after rebuild
  • Locally validated the stress node with temporary request/output/lm-eval limits; aiperf completed 16/16 requests and GSM8K parsed as 0.75 against the 0.42 threshold. Temporary limits are not committed.
  • Pre-merge CI

Summary by CodeRabbit

Release Notes

  • Tests
    • Added disaggregated test coverage for Qwen3-32B-FP8 model with speculative decoding configuration.
    • Improved GSM8K accuracy extraction to support variable whitespace and both integer and decimal values.
    • Extended stress test parametrization to include new Qwen3-32B-FP8 test scenarios on H200 GPUs.

Review Change Stack

@brnguyen2
Copy link
Copy Markdown
Collaborator Author

/bot run --extra-stage "DGX_H200-8_GPUs-PyTorch-Post-Merge-1"

@brnguyen2 brnguyen2 force-pushed the brnguyen/qwen3-32b-fp8-disagg-stress branch from 181a22d to ce5f6a0 Compare May 19, 2026 13:07
@brnguyen2
Copy link
Copy Markdown
Collaborator Author

/bot run --extra-stage "DGX_H200-8_GPUs-PyTorch-Post-Merge-1"

@brnguyen2
Copy link
Copy Markdown
Collaborator Author

/bot help

@github-actions
Copy link
Copy Markdown

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) --high-priority]

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. Supports wildcard * for pattern matching (e.g., "*PerfSanity*" matches all stages containing PerfSanity). Examples: "A10-PyTorch-1, xxx", "PerfSanity". 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. Supports wildcard * for pattern matching. Examples: --extra-stage "H100_PCIe-TensorRT-Post-Merge-1, xxx", --extra-stage "Post-Merge".

--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.

--high-priority (OPTIONAL) : Run the pipeline with high priority. This option is restricted to authorized users only and will route the job to a high-priority queue.

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.

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49195 [ run ] triggered by Bot. Commit: ce5f6a0 Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49195 [ run ] completed with state SUCCESS. Commit: ce5f6a0
/LLM/main/L0_MergeRequest_PR pipeline #38871 completed with status: 'FAILURE'

CI Report

⚠️ Action Required:

  • Please check the failed tests and fix your PR
  • If you cannot view the failures, ask the CI triggerer to share details
  • Once fixed, request an NVIDIA team member to trigger CI again

CI Agent Failure Analysis

Link to invocation

@brnguyen2 brnguyen2 force-pushed the brnguyen/qwen3-32b-fp8-disagg-stress branch 5 times, most recently from c7357ef to e4166bb Compare May 20, 2026 18:55
@brnguyen2
Copy link
Copy Markdown
Collaborator Author

/bot run --extra-stage "DGX_H200-8_GPUs-PyTorch-Post-Merge-1"

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49487 [ run ] triggered by Bot. Commit: e4166bb Link to invocation

@brnguyen2 brnguyen2 force-pushed the brnguyen/qwen3-32b-fp8-disagg-stress branch from e4166bb to 690a201 Compare May 20, 2026 21:10
@brnguyen2
Copy link
Copy Markdown
Collaborator Author

/bot kill

@brnguyen2
Copy link
Copy Markdown
Collaborator Author

/bot run --disable-fail-fast --stage-list "DGX_H200-8_GPUs-PyTorch-Post-Merge-1"

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49493 [ run ] triggered by Bot. Commit: 690a201 Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49494 [ kill ] triggered by Bot. Commit: 690a201 Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49493 [ run ] completed with state ABORTED. Commit: 690a201

Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49487 [ run ] completed with state ABORTED. Commit: e4166bb

Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49494 [ kill ] completed with state SUCCESS. Commit: 690a201
Successfully killed previous jobs for commit 690a201

Link to invocation

@brnguyen2
Copy link
Copy Markdown
Collaborator Author

/bot run --disable-fail-fast --stage-list "DGX_H200-8_GPUs-PyTorch-Post-Merge-1"

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49495 [ run ] triggered by Bot. Commit: 690a201 Link to invocation

@brnguyen2
Copy link
Copy Markdown
Collaborator Author

Verified that the targeted rerun reached the newly added test and it passed:

DGX_H200-8_GPUs-PyTorch-Post-Merge-1/disaggregated/test_disaggregated.py::test_disaggregated_qwen3_32b_fp8[Qwen3/Qwen3-32B-FP8]

The selected test run completed successfully with this pytest summary: 16 passed, 12 skipped, 17175 deselected.

@brnguyen2 brnguyen2 force-pushed the brnguyen/qwen3-32b-fp8-disagg-stress branch from 690a201 to 2c22fef Compare May 21, 2026 00:22
@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49495 [ run ] completed with state SUCCESS. Commit: 690a201
/LLM/main/L0_MergeRequest_PR pipeline #39133 (Partly Tested) completed with status: 'SUCCESS'

CI Report

Link to invocation

@brnguyen2 brnguyen2 marked this pull request as ready for review May 21, 2026 00:28
@brnguyen2 brnguyen2 requested a review from a team as a code owner May 21, 2026 00:28
@brnguyen2
Copy link
Copy Markdown
Collaborator Author

/bot run --disable-fail-fast

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49765 [ kill ] triggered by Bot. Commit: 87fe025 Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49766 [ run ] triggered by Bot. Commit: 87fe025 Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49765 [ kill ] completed with state ABORTED. Commit: 87fe025

Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49751 [ run ] completed with state ABORTED. Commit: ea33e75

Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49766 [ run ] completed with state SUCCESS. Commit: 87fe025
/LLM/main/L0_MergeRequest_PR pipeline #39365 completed with status: 'FAILURE'

CI Report

⚠️ Action Required:

  • Please check the failed tests and fix your PR
  • If you cannot view the failures, ask the CI triggerer to share details
  • Once fixed, request an NVIDIA team member to trigger CI again

CI Agent Failure Analysis

Link to invocation

@brnguyen2
Copy link
Copy Markdown
Collaborator Author

The recent CI failure on this PR (L0_MergeRequest_PR #39365L0_Test-x86_64-Single-GPU #2493, stage [DGX_B200-PyTorch-5]) is an unrelated pre-existing regression on main:

  • Failing test: accuracy/test_llm_api_pytorch.py::TestQwen3_5_35B_A3B::test_bf16_mtp[mtp_on]
  • Error: AttributeError: 'NoneType' object has no attribute 'gather_ids' at tensorrt_llm/_torch/models/modeling_speculative.py:1748 during autotuner warmup.
  • Likely introduced by [TRTLLM-12500][feat] Add support for Qwen3.5 VL MoE - REVERTED by #14599 #14164 (Qwen3.5 VL MoE) — Qwen3.5-35B-A3B text MoE is being routed through modeling_qwen3vl.py and spec_metadata is None during the warmup forward.
  • The mtp_off variant of the same test passes.
  • Tracked in https://nvbugs/6206179.

This PR only touches the Qwen3-32B FP8 disagg stress test plumbing — none of its files are related to modeling_speculative.py / modeling_qwen3vl.py / autotuner warmup. No upstream commit since 96a4a0937e touches those files, so a rebase doesn't help. The latest commit on this branch (473ba2c) waives the failing test pending an upstream fix.

Failure analysis: https://pbss.s8k.io/v1/AUTH_svc_tensorrt/sw-tensorrt-ci-analysis/LLM/main/L0_MergeRequest_PR/39365/failure_analysis.html

@brnguyen2
Copy link
Copy Markdown
Collaborator Author

/bot run

@brnguyen2
Copy link
Copy Markdown
Collaborator Author

/bot kill

@brnguyen2
Copy link
Copy Markdown
Collaborator Author

/bot run --disable-fail-fast

brnguyen2 added 4 commits May 22, 2026 06:46
Add a Qwen3-32B FP8 disaggregated serving smoke and stress test that exercises Eagle3 with 4x TP1 context workers and 1x TP4 generation worker on 8 GPUs.

The YAML enables FP8 KV cache, chunked prefill, block and partial reuse, cache transfer, and a top-level Eagle3 speculative_config shared by context and generation workers. The draft model is stored as a model-root-relative path and the disagg harness now resolves relative model and speculative_model values through llm_models_root while preserving absolute paths.

Wire the smoke test into the H200 L0 list and the full 10k-request stress case into the QA stress list. Add Qwen-specific output substring checks and keep the stress accuracy threshold aligned with the adjacent GPT-OSS stress case.

Signed-off-by: Brian Nguyen <brnguyen@nvidia.com>
Signed-off-by: Brian Nguyen <brnguyen@nvidia.com>
Signed-off-by: Brian Nguyen <brnguyen@nvidia.com>
…t_bf16_mtp[mtp_on]

The test crashes during autotuner warmup with
'NoneType' object has no attribute 'gather_ids' at
modeling_speculative.py:1748 when MTP eagle one-model is combined
with Qwen3.5-35B-A3B. Pre-existing regression on main introduced by
the Qwen3.5 VL MoE landing (96a4a09); unrelated to this PR's
changes. Tracked in https://nvbugs/6206179.

Signed-off-by: Brian Nguyen <brnguyen@nvidia.com>
@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49929 [ run ] triggered by Bot. Commit: 473ba2c Link to invocation

@brnguyen2 brnguyen2 force-pushed the brnguyen/qwen3-32b-fp8-disagg-stress branch from 473ba2c to 02a618a Compare May 22, 2026 11:46
@brnguyen2
Copy link
Copy Markdown
Collaborator Author

/bot run --disable-fail-fast

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49930 [ kill ] triggered by Bot. Commit: 02a618a Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49929 [ run ] completed with state ABORTED. Commit: 473ba2c

Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49930 [ kill ] completed with state SUCCESS. Commit: 02a618a
Successfully killed previous jobs for commit 02a618a

Link to invocation

@brnguyen2
Copy link
Copy Markdown
Collaborator Author

/bot kill

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49931 [ run ] triggered by Bot. Commit: 02a618a Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49933 [ run ] triggered by Bot. Commit: 02a618a Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49934 [ kill ] triggered by Bot. Commit: 02a618a Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49933 [ run ] completed with state ABORTED. Commit: 02a618a

Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49931 [ run ] completed with state ABORTED. Commit: 02a618a

Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49934 [ kill ] completed with state SUCCESS. Commit: 02a618a
Successfully killed previous jobs for commit 02a618a

Link to invocation

@brnguyen2
Copy link
Copy Markdown
Collaborator Author

/bot run --disable-fail-fast

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49935 [ run ] triggered by Bot. Commit: 02a618a Link to invocation

@tensorrt-cicd
Copy link
Copy Markdown
Collaborator

PR_Github #49935 [ run ] completed with state SUCCESS. Commit: 02a618a
/LLM/main/L0_MergeRequest_PR pipeline #39508 completed with status: 'FAILURE'

CI Report

⚠️ Action Required:

  • Please check the failed tests and fix your PR
  • If you cannot view the failures, ask the CI triggerer to share details
  • Once fixed, request an NVIDIA team member to trigger CI again

CI Agent Failure Analysis

Link to invocation

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants