Skip to content

Commit

Permalink
Fix logs
Browse files Browse the repository at this point in the history
  • Loading branch information
tanmayv25 committed Sep 12, 2023
1 parent cfd9d23 commit a631b77
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 6 deletions.
5 changes: 2 additions & 3 deletions src/grpc/infer_handler.cc
Original file line number Diff line number Diff line change
Expand Up @@ -943,12 +943,11 @@ ModelInferHandler::Execute(InferHandler::State* state)
// If not error then state->step_ == ISSUED and inference request
// has initiated... completion callback will transition to
// COMPLETE or CANCELLED. Recording the state and the irequest
// to handle gRPC stream cancellation. If error go immediately to
// COMPLETE.
// to handle gRPC stream cancellation.
if (err == nullptr) {
state->context_->InsertInflightState(state, irequest);
} else {
// Handling error...
// If error go immediately to COMPLETE.
LOG_VERBOSE(1) << "[request id: " << request_id << "] "
<< "Infer failed: " << TRITONSERVER_ErrorMessage(err);

Expand Down
6 changes: 3 additions & 3 deletions src/grpc/stream_infer_handler.cc
Original file line number Diff line number Diff line change
Expand Up @@ -310,12 +310,12 @@ ModelStreamInferHandler::Process(InferHandler::State* state, bool rpc_ok)
// state->step_ == ISSUED and inference request has
// initiated... the completion callback will transition to
// WRITEREADY or WRITTEN or CANCELLED. Recording the state and the
// irequest to handle gRPC stream cancellation. If there was an
// error then enqueue the error response and show it to be ready
// for writing.
// irequest to handle gRPC stream cancellation.
if (err == nullptr) {
state->context_->InsertInflightState(state, irequest);
} else {
// If there was an error then enqueue the error response and show
// it to be ready for writing.
inference::ModelStreamInferResponse* response;
if (state->is_decoupled_) {
state->response_queue_->AllocateResponse();
Expand Down

0 comments on commit a631b77

Please sign in to comment.