Add double parameter handling (#333)

* Support Double-Type Infer/Response Parameters
triton-inference-server · Feb 1, 2024 · 0371eb8 · 0371eb8
1 parent 37d2902
commit 0371eb8
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,