File tree Expand file tree Collapse file tree 1 file changed +5
-6
lines changed
src/sequence_batch_scheduler Expand file tree Collapse file tree 1 file changed +5
-6
lines changed Original file line number Diff line number Diff 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}
You can’t perform that action at this time.
0 commit comments