From b0b369d894642f65440a6b3066d6f21ea657d96b Mon Sep 17 00:00:00 2001 From: fpetrini15 Date: Fri, 29 Dec 2023 16:50:57 -0800 Subject: [PATCH] Add double parameter handling --- src/python_be.cc | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/python_be.cc b/src/python_be.cc index 3c9dd19d..e3cafe61 100644 --- 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(vvalue); RETURN_IF_ERROR(parameters_json.AddString(name, string)); + } else if (type == TRITONSERVER_PARAMETER_DOUBLE) { + RETURN_IF_ERROR(parameters_json.AddDouble( + name, *(reinterpret_cast(vvalue)))); } else { return TRITONSERVER_ErrorNew( TRITONSERVER_ERROR_INVALID_ARG,