nnshah1 stream infer segfault fix (#5842)

match logic from infer_handler.cc
triton-inference-server · May 23, 2023 · aeccc45 · aeccc45
1 parent c1ad1bf
commit aeccc45
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/grpc/stream_infer_handler.cc b/src/grpc/stream_infer_handler.cc
@@ -269,9 +269,11 @@ ModelStreamInferHandler::Process(InferHandler::State* state, bool rpc_ok)
     }
     // Get request ID for logging in case of error.
     const char* request_id = "";
-    LOG_TRITONSERVER_ERROR(
-        TRITONSERVER_InferenceRequestId(irequest, &request_id),
-        "unable to retrieve request ID string");
+    if (irequest != nullptr) {
+      LOG_TRITONSERVER_ERROR(
+          TRITONSERVER_InferenceRequestId(irequest, &request_id),
+          "unable to retrieve request ID string");
+    }
     if (!strncmp(request_id, "", 1)) {
       request_id = "<id_unknown>";
     }