Improve error message

triton-inference-server · Jul 19, 2024 · 5b8c263 · 5b8c263
1 parent 8b5e3ba
commit 5b8c263
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/instance_state.cc b/src/instance_state.cc
@@ -778,9 +778,9 @@ ModelInstanceState::Run(
               payload_->responses_,
               TRITONSERVER_ErrorNew(
                   TRITONSERVER_ERROR_INVALID_ARG,
-                  (std::string("tensor for input '") + name +
-                   "' expected byte size is " +
-                   std::to_string(total_byte_size) + ", got " +
+                  (std::string("input byte size mismatch for input '") + name +
+                   "'" + " for model '" + model_state_->Name() +
+                   "'. Expected " + std::to_string(total_byte_size) + ", got " +
                    std::to_string(req_data_byte_size))
                       .c_str()),
               "failed to run TRT inference");