Assert on json input format

triton-inference-server · Mar 5, 2024 · 0e06c11 · 0e06c11
1 parent 20cc90b
commit 0e06c11
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/c++/perf_analyzer/perf_utils.cc b/src/c++/perf_analyzer/perf_utils.cc
@@ -205,7 +205,13 @@ SerializeExplicitTensor(
   } else if (dt.compare("JSON") == 0) {
     std::string serialized = "";
 
-    for (const auto& value : tensor.GetArray()) {
+    auto values = tensor.GetArray();
+    if (values.Size() != 1) {
+      return cb::Error(
+          "JSON format does not yet support multiple json objects in the "
+          "input");
+    }
+    for (const auto& value : values) {
       rapidjson::StringBuffer buffer;
       rapidjson::Writer<rapidjson::StringBuffer> writer(buffer);
       value.Accept(writer);