Skip to content

[BugFix] Fix several bugs in the request interruption and inference termination functionality#6278

Open
qwes5s5 wants to merge 2 commits intoPaddlePaddle:developfrom
qwes5s5:fix_request_disconnect
Open

[BugFix] Fix several bugs in the request interruption and inference termination functionality#6278
qwes5s5 wants to merge 2 commits intoPaddlePaddle:developfrom
qwes5s5:fix_request_disconnect

Conversation

@qwes5s5
Copy link
Collaborator

@qwes5s5 qwes5s5 commented Jan 29, 2026

Motivation

💡 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)

Current request interruption for inference termination has issues in both centralized and PD-disaggregated modes. These bugs can potentially cause resource leakage, service instability, and request hangs.

Modifications

  • Construct and dispatch ScheduledAbortTask to workers to distinguish it from ScheduledPreemptTask.
  • Implement differential handling based on the request's role and state within the ResourceManager to mitigate resource leaks.
  • Optimize the overall performance of inference interruption in PD-separated mode, ensuring requests are interrupted and resources are reclaimed across P-nodes, D-nodes, and during transmission.

Usage or Command

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

paddle-bot bot commented Jan 29, 2026

Thanks for your contribution!

@codecov-commenter
Copy link

codecov-commenter commented Jan 29, 2026

Codecov Report

❌ Patch coverage is 35.59322% with 114 lines in your changes missing coverage. Please review.
⚠️ Please upload report for BASE (develop@c4abb01). Learn more about missing BASE report.

Files with missing lines Patch % Lines
fastdeploy/engine/common_engine.py 16.96% 90 Missing and 3 partials ⚠️
fastdeploy/splitwise/splitwise_connector.py 0.00% 9 Missing ⚠️
fastdeploy/scheduler/local_scheduler.py 40.00% 5 Missing and 1 partial ⚠️
fastdeploy/engine/sched/resource_manager_v1.py 85.71% 0 Missing and 3 partials ⚠️
fastdeploy/scheduler/dp_scheduler.py 80.00% 1 Missing and 1 partial ⚠️
fastdeploy/output/token_processor.py 90.90% 0 Missing and 1 partial ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##             develop    #6278   +/-   ##
==========================================
  Coverage           ?   67.57%           
==========================================
  Files              ?      391           
  Lines              ?    52867           
  Branches           ?     8244           
==========================================
  Hits               ?    35727           
  Misses             ?    14523           
  Partials           ?     2617           
Flag Coverage Δ
GPU 67.57% <35.59%> (?)

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.

@qwes5s5 qwes5s5 changed the title [BugFix] Fix several bugs in PR #5320 [BugFix] Fix several bugs in the request interruption and inference termination functionality Feb 11, 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

Comments