Skip to content

Commit fc2ada2

Browse files
authored
updating log message with input name
1 parent e456d2a commit fc2ada2

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

src/infer_request.cc

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -807,7 +807,9 @@ InferenceRequest::Normalize()
807807
if (!has_one_element) {
808808
return Status(
809809
Status::Code::INVALID_ARG, LogRequest() +
810-
"For BYTE datatype raw input, the "
810+
"For BYTE datatype raw input '" +
811+
config_input.name() +
812+
"', the "
811813
"model must have input shape [1]");
812814
}
813815
// In the case of BYTE data type, we will prepend the byte size to follow
@@ -946,13 +948,13 @@ InferenceRequest::Normalize()
946948
if (input.DType() != input_config->data_type()) {
947949
return Status(
948950
Status::Code::INVALID_ARG,
949-
LogRequest() + "inference input data-type is '" +
951+
LogRequest() + "inference input '" + pr.first + "' data-type is '" +
950952
std::string(
951953
triton::common::DataTypeToProtocolString(input.DType())) +
952-
"', model expects '" +
954+
"', but model '" + ModelName() + "' expects '" +
953955
std::string(triton::common::DataTypeToProtocolString(
954956
input_config->data_type())) +
955-
"' for '" + ModelName() + "'");
957+
"'");
956958
}
957959

958960
// Validate input shape

0 commit comments

Comments
 (0)