Skip to content

Commit 0371eb8

Browse files
authored
Add double parameter handling (triton-inference-server#333)
* Support Double-Type Infer/Response Parameters
1 parent 37d2902 commit 0371eb8

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/python_be.cc

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -340,6 +340,9 @@ ModelInstanceState::SaveRequestsToSharedMemory(
340340
} else if (type == TRITONSERVER_PARAMETER_STRING) {
341341
std::string string = reinterpret_cast<const char*>(vvalue);
342342
RETURN_IF_ERROR(parameters_json.AddString(name, string));
343+
} else if (type == TRITONSERVER_PARAMETER_DOUBLE) {
344+
RETURN_IF_ERROR(parameters_json.AddDouble(
345+
name, *(reinterpret_cast<const double*>(vvalue))));
343346
} else {
344347
return TRITONSERVER_ErrorNew(
345348
TRITONSERVER_ERROR_INVALID_ARG,

0 commit comments

Comments
 (0)