updating log message with input name

triton-inference-server · Aug 18, 2023 · fc2ada2 · fc2ada2
1 parent e456d2a
commit fc2ada2
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/infer_request.cc b/src/infer_request.cc
@@ -807,7 +807,9 @@ InferenceRequest::Normalize()
       if (!has_one_element) {
         return Status(
             Status::Code::INVALID_ARG, LogRequest() +
-                                           "For BYTE datatype raw input, the "
+                                           "For BYTE datatype raw input '" +
+                                           config_input.name() +
+                                           "', the "
                                            "model must have input shape [1]");
       }
       // In the case of BYTE data type, we will prepend the byte size to follow
@@ -946,13 +948,13 @@ InferenceRequest::Normalize()
     if (input.DType() != input_config->data_type()) {
       return Status(
           Status::Code::INVALID_ARG,
-          LogRequest() + "inference input data-type is '" +
+          LogRequest() + "inference input '" + pr.first + "' data-type is '" +
               std::string(
                   triton::common::DataTypeToProtocolString(input.DType())) +
-              "', model expects '" +
+              "', but model '" + ModelName() + "' expects '" +
               std::string(triton::common::DataTypeToProtocolString(
                   input_config->data_type())) +
-              "' for '" + ModelName() + "'");
+              "'");
     }
 
     // Validate input shape