Skip to content

[CI][ROCm][Disagg] Enable breakable CUDA graphs for MiniMax-M3-MXFP8 decode - #55829

Closed
avininjamay8 wants to merge 1 commit into
vllm-project:mainfrom
avininjamay8:fix/mm3-breakable-cudagraph
Closed

avininjamay8 wants to merge 1 commit into
vllm-project:mainfrom
avininjamay8:fix/mm3-breakable-cudagraph

Conversation

@avininjamay8

@avininjamay8 avininjamay8 commented Sep 8, 2026

Copy link
Copy Markdown
Contributor

Purpose

Follow-up to #54782. After that merge, MiniMax-M3-MXFP8 decode fails in our ROCm disagg CI:

RuntimeError: piecewise CUDA graphs (cudagraph_mode=FULL_AND_PIECEWISE) unavailable ...
Set VLLM_USE_BREAKABLE_CUDAGRAPH=1

Prefill runs eager; decode uses piecewise graphs. Set VLLM_USE_BREAKABLE_CUDAGRAPH: "1" in models.yaml so bring-up and GSM8K pass on nightly.

Changes

.buildkite/amd-disagg/models.yaml only:

  • MiniMax-M3-MXFP8 env: VLLM_USE_BREAKABLE_CUDAGRAPH: "0""1"

No launcher, MoRIIO, or harness script changes.

Test Plan

MoRIIO 1P1D TP8 disaggregated serving, ROUTER_TYPE=proxy, RUN_AFTER_HEALTH=accuracy, image vllm/vllm-openai-rocm:nightly.

MiniMax-M3-MXFP8 — 1P1D TP8

export VLLM_ROCM_USE_AITER=1
export VLLM_ROCM_USE_AITER_MOE=1
export VLLM_ROCM_USE_AITER_RMSNORM=1
export VLLM_USE_BREAKABLE_CUDAGRAPH=1
export VLLM_ROCM_QUICK_REDUCE_QUANTIZATION=INT6

# PREFILL — node0
vllm serve /data/models2/MiniMax-M3-MXFP8 \
    --host $PREFILL_IP --port 8100 -tp 8 \
    --trust-remote-code \
    --attention-backend TRITON_ATTN \
    --block-size 128 \
    --language-model-only \
    --kv-cache-dtype fp8 \
    --gpu-memory-utilization 0.85 \
    --enforce-eager \
    --kv-transfer-config '{"kv_connector":"MoRIIOConnector","kv_role":"kv_producer","kv_port":"14579",
      "kv_connector_extra_config":{"proxy_ip":"'$PREFILL_IP'","proxy_port":"10001",
      "proxy_ping_port":"36367","http_port":"8100","local_ping_port":"61555",
      "handshake_port":"6301","notify_port":"61005"}}'

# DECODE — node1
vllm serve /data/models2/MiniMax-M3-MXFP8 \
    --host $DECODE_IP --port 8200 -tp 8 \
    --trust-remote-code \
    --attention-backend TRITON_ATTN \
    --block-size 128 \
    --language-model-only \
    --kv-cache-dtype fp8 \
    --gpu-memory-utilization 0.85 \
    --kv-transfer-config '{"kv_connector":"MoRIIOConnector","kv_role":"kv_consumer","kv_port":"14579",
      "kv_connector_extra_config":{"proxy_ip":"'$PREFILL_IP'","proxy_port":"10001",
      "proxy_ping_port":"36367","http_port":"8200","local_ping_port":"61555",
      "handshake_port":"6301","notify_port":"61005"}}'

GSM8K evaluation

lm_eval --model local-completions \
    --tasks gsm8k \
    --model_args "model=/data/models2/MiniMax-M3-MXFP8,base_url=http://127.0.0.1:10001/v1/completions,num_concurrent=64,max_retries=3,tokenized_requests=False,trust_remote_code=True,timeout=7200" \
    --limit 250

Test Result

MiniMax-M3-MXFP8, MoRIIO 1P1D TP8 (proxy). GSM8K flexible-extract, threshold 0.90.

Variant Health GSM8K exact_match
#54782 + upstream models.yaml (VLLM_USE_BREAKABLE_CUDAGRAPH=0) Fail (decode piecewise CUDA graphs unavailable)
#54782 + this fix Pass 0.9128 (PASS)

Nightly vLLM (vllm-project#54782) now requires breakable CUDA graphs or eager mode
for MiniMax sparse decode when piecewise graphs are enabled. Set
VLLM_USE_BREAKABLE_CUDAGRAPH=1 so decode bring-up passes on nightly.

Co-authored-by: Cursor <cursoragent@cursor.com>
@mergify mergify Bot added the rocm Related to AMD ROCm label Sep 8, 2026
@github-project-automation github-project-automation Bot moved this to Todo in AMD Sep 8, 2026
@github-project-automation github-project-automation Bot moved this from Todo to Done in AMD Sep 8, 2026
@github-project-automation github-project-automation Bot moved this to Done in NVIDIA Sep 8, 2026
@avininjamay8 avininjamay8 reopened this Sep 8, 2026
@avininjamay8 avininjamay8 reopened this Sep 8, 2026
@avininjamay8 avininjamay8 reopened this Sep 8, 2026
@avininjamay8

Copy link
Copy Markdown
Contributor Author

@lcskrishna were you mentioning abt this issue in Thursday's evening meeting , will close this incase you addressed already.

@avininjamay8

Copy link
Copy Markdown
Contributor Author

Closing stuck draft PR; reopening as ready-for-review (non-draft).

@avininjamay8 avininjamay8 reopened this Sep 9, 2026
@avininjamay8

Copy link
Copy Markdown
Contributor Author

Closing stuck draft; replacing with ready-for-review PR from fix/mm3-breakable-cudagraph-open.

@avininjamay8

Copy link
Copy Markdown
Contributor Author

Superseded by #56075 (non-draft replacement).

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

Labels

Projects

Status: Done
Status: Done

Development

Successfully merging this pull request may close these issues.

1 participant