Add double parameter handling

triton-inference-server · Dec 30, 2023 · b0b369d · b0b369d
1 parent 950c47f
commit b0b369d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/python_be.cc b/src/python_be.cc
@@ -340,6 +340,9 @@ ModelInstanceState::SaveRequestsToSharedMemory(
       } else if (type == TRITONSERVER_PARAMETER_STRING) {
         std::string string = reinterpret_cast<const char*>(vvalue);
         RETURN_IF_ERROR(parameters_json.AddString(name, string));
+      } else if (type == TRITONSERVER_PARAMETER_DOUBLE) {
+        RETURN_IF_ERROR(parameters_json.AddDouble(
+            name, *(reinterpret_cast<const double*>(vvalue))));
       } else {
         return TRITONSERVER_ErrorNew(
             TRITONSERVER_ERROR_INVALID_ARG,