From a3685040bacbfff109b7505b5f2f182122253366 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 a8dfab07..befdd593 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,