Skip to content

[XPU] Enable QK Norm + RoPE fusion pass on XPU - #49394

Merged
jikunshang merged 7 commits into
vllm-project:mainfrom
chaojun-zhang:fix/xpu-qk-norm-rope-fusion
Jul 27, 2026
Merged

[XPU] Enable QK Norm + RoPE fusion pass on XPU#49394
jikunshang merged 7 commits into
vllm-project:mainfrom
chaojun-zhang:fix/xpu-qk-norm-rope-fusion

Conversation

@chaojun-zhang

@chaojun-zhang chaojun-zhang commented Jul 22, 2026

Copy link
Copy Markdown
Contributor

Performance & Accuracy Test

Commands

Server (fusion_on adds --compilation-config or set , eager adds --enforce-eager instead):

vllm serve Qwen/Qwen3-8B --port 8072 --host 0.0.0.0 --trust-remote-code \
    --gpu-memory-util=0.9 --no-enable-prefix-caching \
    --max-num-batched-tokens=8192 --max-model-len=6000 --block-size 64 -tp=1 \
    --compilation-config '{"pass_config":{"enable_qk_norm_rope_fusion":true}}'

Bench client:

vllm bench serve --model Qwen/Qwen3-8B --dataset-name random \
    --random-input-len=4096 --random-output-len=1024 --ignore-eos \
    --num-prompt 32 --max-concurrency 8 --request-rate inf --backend vllm \
    --trust-remote-code --temperature=0 --port 8072

Benchmark report (input=4096, output=1024, num-prompts=32, max-concurrency=8)

Case Duration (s) Output tok/s Total tok/s Mean TTFT (ms) Mean TPOT (ms)
fusion_on 321.33 101.98 509.89 17651.13 53.39
fusion_off 358.83 91.32 456.60 23016.48 54.09
eager[baseline] 321.58 101.90 509.48 16938.86 55.13

fusion_on gives +11.7% throughput and -10.5% duration vs fusion_off, matching eager.

Accuracy report (GSM8K, 5-shot, limit=250)

Case Exact match (strict) Exact match (flexible)
fusion_on 0.908 0.916
fusion_off 0.912 0.916
eager 0.904 0.912

All within 1 stderr of each other — no accuracy regression from the fusion pass.

@claude claude Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

Claude Code Review

This pull request is from a fork — automated review is disabled. A repository maintainer can comment @claude review to run a one-time review.

@mergify mergify Bot added the intel-gpu Related to Intel GPU label Jul 22, 2026
Signed-off-by: Chaojun Zhang <chaojun.zhang@intel.com>
@chaojun-zhang
chaojun-zhang force-pushed the fix/xpu-qk-norm-rope-fusion branch from 16086b8 to b9bff44 Compare July 22, 2026 04:18
@mergify mergify Bot added the ci/build label Jul 22, 2026

@jikunshang jikunshang left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

@jikunshang jikunshang added the ready ONLY add when PR is ready to merge/full CI is needed label Jul 22, 2026
@jikunshang
jikunshang enabled auto-merge (squash) July 26, 2026 14:22
@jikunshang
jikunshang merged commit bf4f633 into vllm-project:main Jul 27, 2026
92 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ci/build intel-gpu Related to Intel GPU ready ONLY add when PR is ready to merge/full CI is needed

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants