Skip to content

[Feat][1/2] Add Transformers TPSP dense baseline support - #62385

Draft
hlin99 wants to merge 3 commits into
vllm-project:mainfrom
hlin99:ww43_PR_TPSP_Transformers_1/2
Draft

hlin99 wants to merge 3 commits into
vllm-project:mainfrom
hlin99:ww43_PR_TPSP_Transformers_1/2

Conversation

@hlin99

@hlin99 hlin99 commented Oct 19, 2026 •

Copy link
Copy Markdown
Contributor

Design doc: https://github.com/hlin99/vllm/blob/ww43_PR_TPSP_Transformers_1/2/docs/design/tpsp_design.md

Verified model list

  • Qwen/Qwen3-8B
  • Qwen/Qwen2-7B-Instruct
  • tencent/Hunyuan-7B-Instruct
  • zai-org/glm-4-9b-chat-hf
  • meta-llama/Meta-Llama-3.1-8B-Instruct
  • baidu/ERNIE-4.5-0.3B-PT
  • mistralai/Mistral-7B-Instruct-v0.1
  • mistralai/Ministral-8B-Instruct-2410
  • HuggingFaceTB/SmolLM3-3B
  • arcee-ai/AFM-4.5B-Preview
  • kyutai/helium-1-preview-2b
  • facebook/cwm

Performance

Delivers up to a ~25% boost in E2E TTFT under optimal conditions, subject to individual device compute and communication bottlenecks.

image image

Profiles

#1 PCIE Communication Topology (Communicationas the Long Pole vs. GEMM)

By splitting a slow All-Reduce into fast peer-to-peer (P2P) intra-node copies, we successfully mitigated the All-Reduce bottleneck and overlapped GEMM computations within the communication phase (except for the first and last micro-chunks). This resulted in a ~40% performance gain, as demonstrated in the down_proj + norm example below.

Figure#1.A. TPSP off

image

Figure#1.B. TPSP on

image

#2 NVLink Communication Topology (GEMM as the Long Pole vs. Communication)

By tuning SM allocation and chunk size, we successfully split All-Reduce and normalization into Reduce-Scatter + All-Gather + Normalization (RS + AG + Norm), allowing them to overlap seamlessly with GEMM computations(except for the first & last micro chunk). This resulted in a ~8% performance gain, as demonstrated in the down_proj + norm example below.

Figure#2.A. TPSP off

image

Figure#2.B. TPSP on

image

@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 commented Oct 19, 2026

Copy link
Copy Markdown
Contributor

Documentation preview: https://vllm--62385.org.readthedocs.build/en/62385/

@mergify mergify Bot added documentation Improvements or additions to documentation ci/build nvidia labels Oct 19, 2026
@mergify

mergify Bot commented Oct 19, 2026

Copy link
Copy Markdown
Contributor

This pull request has merge conflicts that must be resolved before it can be
merged. Please rebase the PR, @hlin99.

https://docs.github.com/en/pull-requests/collaborating-with-pull-requests/working-with-forks/syncing-a-fork

@mergify mergify Bot added the needs-rebase label Oct 19, 2026
Signed-off-by: Tony Lin <tony.lin@intel.com>
Signed-off-by: Tony Lin <tony.lin@intel.com>
Signed-off-by: Tony Lin <tony.lin@intel.com>
@mergify

mergify Bot commented Oct 21, 2026

Copy link
Copy Markdown
Contributor

This pull request has merge conflicts that must be resolved before it can be
merged. Please rebase the PR, @hlin99.

https://docs.github.com/en/pull-requests/collaborating-with-pull-requests/working-with-forks/syncing-a-fork

@mergify mergify Bot added the needs-rebase label Oct 21, 2026
@hlin99
hlin99 marked this pull request as draft November 12, 2026 04:37

This branch has not been deployed

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

Labels

ci/build documentation Improvements or additions to documentation needs-rebase nvidia

Projects

Status: No status

Development

Successfully merging this pull request may close these issues.

1 participant