Skip to content

Commit a534580

Browse files
authored
Add release callback when the request is put into the backlog (#337)
1 parent 434e503 commit a534580

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

src/sequence_batch_scheduler/sequence_batch_scheduler.cc

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -711,6 +711,11 @@ SequenceBatchScheduler::Enqueue(std::unique_ptr<InferenceRequest>& irequest)
711711
auto sb_itr = sequence_to_batcherseqslot_map_.find(correlation_id);
712712
auto bl_itr = sequence_to_backlog_map_.find(correlation_id);
713713

714+
sequencer_->AddReleaseCallback(
715+
irequest,
716+
[this](std::unique_ptr<InferenceRequest>& request, const uint32_t flags)
717+
-> Status { return sequencer_->RescheduleRequest(request, flags); });
718+
714719
// If this request is not starting a new sequence its correlation ID
715720
// should already be known with a target in either a sequence slot
716721
// or in the backlog. If it doesn't then the sequence wasn't started
@@ -850,12 +855,6 @@ SequenceBatchScheduler::Enqueue(std::unique_ptr<InferenceRequest>& irequest)
850855
LOG_VERBOSE(1) << "Enqueuing CORRID " << correlation_id << " into batcher "
851856
<< model_instance->Name() << ", sequence slot " << seq_slot
852857
<< ": " << irequest->ModelName();
853-
854-
sequencer_->AddReleaseCallback(
855-
irequest,
856-
[this](std::unique_ptr<InferenceRequest>& request, const uint32_t flags)
857-
-> Status { return sequencer_->RescheduleRequest(request, flags); });
858-
859858
batchers_[model_instance]->Enqueue(seq_slot, correlation_id, irequest);
860859
return Status::Success;
861860
}

0 commit comments

Comments
 (0)