Skip to content

[Feature][OP] Append Attn Support CUDA-PDL - #5072

Merged
zhoutianzi666 merged 6 commits into
PaddlePaddle:developfrom
ckl117:dev_append_attn_pdl
Nov 17, 2025
Merged

zhoutianzi666 merged 6 commits into
PaddlePaddle:developfrom
ckl117:dev_append_attn_pdl

Conversation

@ckl117

@ckl117 ckl117 commented Nov 16, 2025

Copy link
Copy Markdown
Collaborator

Motivation

Before enabling PDL, there is a bubble between two kernels.
image

After enabling PDL, there is almost no bubbles between kernels.
image

Modifications

  • A new env FD_ENABLE_PDL, with a default value of 1, which means CUDA-PDL is enabled when sm>=90.
  • Modified the startup method of the kernel.
  • Changed some kernel names (when input's dtype is int).

Usage or Command

export FD_ENABLE_PDL=0 will disable CUDA-PDL.

Accuracy Tests

Append Attn Support CUDA-PDL

Checklist

  • Add at least a tag in the PR title.
    • Tag list: [[FDConfig],[APIServer],[Engine], [Scheduler], [PD Disaggregation], [Executor], [Graph Optimization], [Speculative Decoding], [RL], [Models], [Quantization], [Loader], [OP], [KVCache], [DataProcessor], [BugFix], [Docs], [CI], [Optimization], [Feature], [Benchmark], [Others], [XPU], [HPU], [GCU], [DCU], [Iluvatar], [Metax]]
    • You can add new tags based on the PR content, but the semantics must be clear.
  • Format your code, run pre-commit before commit.
  • Add unit tests. Please write the reason in this PR if no unit tests.
  • Provide accuracy results.
  • If the current PR is submitting to the release branch, make sure the PR has been submitted to the develop branch, then cherry-pick it to the release branch with the [Cherry-Pick] PR tag.

@paddle-bot

paddle-bot Bot commented Nov 16, 2025

Copy link
Copy Markdown

Thanks for your contribution!

@ckl117 ckl117 changed the title [OP] Append Attn Support CUDA-PDL [Feature][OP] Append Attn Support CUDA-PDL Nov 17, 2025
@zhoutianzi666
zhoutianzi666 merged commit d58c1db into PaddlePaddle:develop Nov 17, 2025
14 of 16 checks passed
chang-wenbin pushed a commit to chang-wenbin/FastDeploy that referenced this pull request Mar 2, 2026
xiaoguoguo626807 pushed a commit to xiaoguoguo626807/FastDeploy that referenced this pull request May 7, 2026
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.

2 participants