Skip to content

[Feature] Support prefill batch inference for pooling models.#5436

Merged
Jiang-Jia-Jun merged 11 commits into
PaddlePaddle:developfrom
lizexu123:support_pooling_11
Dec 9, 2025
Merged

[Feature] Support prefill batch inference for pooling models.#5436
Jiang-Jia-Jun merged 11 commits into
PaddlePaddle:developfrom
lizexu123:support_pooling_11

Conversation

@lizexu123
Copy link
Copy Markdown
Collaborator

@lizexu123 lizexu123 commented Dec 8, 2025

Motivation

支持pooling模型多batch推理

💡 If this PR is a Cherry Pick, the PR title needs to follow the format by adding the [Cherry-Pick] label at the very beginning and appending the original PR ID at the end. For example, [Cherry-Pick][CI] Add check trigger and logic(#5191)

💡 如若此PR是Cherry Pick,PR标题需遵循格式,在最开始加上[Cherry-Pick]标签,以及最后面加上原PR ID,例如[Cherry-Pick][CI] Add check trigger and logic(#5191)

Modifications

Usage or Command

对于纯文多batch推理

export FD_ENABLE_MAX_PREFILL=1

model_path=Qwen3-Embedding-0.6B
python -m fastdeploy.entrypoints.openai.api_server \
    --model ${model_path} \
    --max-num-seqs 256 \
    --max-model-len 32768 \
    --port 8070 \
    --engine-worker-queue-port 7562 \
    --metrics-port 7531 \
    --tensor-parallel-size 1 \
    --gpu-memory-utilization 0.9 \
    --runner pooling \
    --convert embed \
    --max-num-partial-prefills 10 \
    --max-long-partial-prefills 10 \

Accuracy Tests

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
Copy link
Copy Markdown

paddle-bot Bot commented Dec 8, 2025

Thanks for your contribution!

@lizexu123 lizexu123 changed the title [Feature] Support multi-batch inference for pooling models. [Feature] Support prefill batch inference for pooling models. Dec 8, 2025
@codecov-commenter
Copy link
Copy Markdown

codecov-commenter commented Dec 8, 2025

Codecov Report

✅ All modified and coverable lines are covered by tests.
⚠️ Please upload report for BASE (develop@438c9f7). Learn more about missing BASE report.

Additional details and impacted files
@@            Coverage Diff             @@
##             develop    #5436   +/-   ##
==========================================
  Coverage           ?   59.94%           
==========================================
  Files              ?      329           
  Lines              ?    40958           
  Branches           ?     6207           
==========================================
  Hits               ?    24552           
  Misses             ?    14539           
  Partials           ?     1867           
Flag Coverage Δ
GPU 59.94% <100.00%> (?)

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@lizexu123 lizexu123 closed this Dec 9, 2025
@lizexu123 lizexu123 reopened this Dec 9, 2025
yuanlehome
yuanlehome previously approved these changes Dec 9, 2025
@Jiang-Jia-Jun Jiang-Jia-Jun merged commit b0cf2c4 into PaddlePaddle:develop Dec 9, 2025
14 of 18 checks passed
chang-wenbin pushed a commit to chang-wenbin/FastDeploy that referenced this pull request Mar 2, 2026
…Paddle#5436)

* fix multi-inputs

* fix threshold

* fix threshold

* fix

* support multi-batch

* add tests

* fix test

* test

* fix
xiaoguoguo626807 pushed a commit to xiaoguoguo626807/FastDeploy that referenced this pull request May 7, 2026
…Paddle#5436)

* fix multi-inputs

* fix threshold

* fix threshold

* fix

* support multi-batch

* add tests

* fix test

* test

* fix
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