Skip to content

Commit 50a3a08

Browse files
authored
bugfix: skip cancelled requests when processing stream output. (#289)
1 parent 9c62060 commit 50a3a08

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

xllm/core/scheduler/continuous_scheduler.cpp

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1019,8 +1019,13 @@ void ContinuousScheduler::process_batch_output(bool enable_schedule_overlap) {
10191019
for (auto request : to_be_processed_requests) {
10201020
// ignore cancelled/finished requests when enable_schedule_overlap.
10211021
if (options_.enable_schedule_overlap() && request->state().stream) {
1022-
if (!request->finished() && !request->cancelled()) {
1022+
// skip cancelled request
1023+
if (request->cancelled()) {
1024+
continue;
1025+
}
1026+
if (!request->finished()) {
10231027
stream_requests.emplace_back(request);
1028+
continue;
10241029
}
10251030
// handle token when last token not be handled.
10261031
if (request->finished() && !request->last_token_handled()) {

0 commit comments

Comments
 (0)