|
1 | | -// Copyright 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved. |
| 1 | +// Copyright 2022-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved. |
2 | 2 | // |
3 | 3 | // Redistribution and use in source and binary forms, with or without |
4 | 4 | // modification, are permitted provided that the following conditions |
@@ -368,8 +368,8 @@ struct ResponseParameters { |
368 | 368 | // The name of the parameter. |
369 | 369 | const char* name_; |
370 | 370 | // The type of the parameter. Valid types are TRITONSERVER_PARAMETER_STRING, |
371 | | - // TRITONSERVER_PARAMETER_INT, TRITONSERVER_PARAMETER_BOOL, and |
372 | | - // TRITONSERVER_PARAMETER_BYTES. |
| 371 | + // TRITONSERVER_PARAMETER_INT, TRITONSERVER_PARAMETER_BOOL, |
| 372 | + // TRITONSERVER_PARAMETER_BYTES and TRITONSERVER_PARAMETER_DOUBLE. |
373 | 373 | TRITONSERVER_ParameterType type_; |
374 | 374 | // The pointer to the parameter value. |
375 | 375 | const void* vvalue_; |
@@ -2057,6 +2057,11 @@ InferResult::DebugString() |
2057 | 2057 | params_[i]->name_, |
2058 | 2058 | reinterpret_cast<const char*>(params_[i]->vvalue_))); |
2059 | 2059 | break; |
| 2060 | + case TRITONSERVER_PARAMETER_DOUBLE: |
| 2061 | + THROW_IF_TRITON_ERR(params_json.AddDouble( |
| 2062 | + params_[i]->name_, |
| 2063 | + *(reinterpret_cast<const double*>(params_[i]->vvalue_)))); |
| 2064 | + break; |
2060 | 2065 | case TRITONSERVER_PARAMETER_BYTES: |
2061 | 2066 | throw TritonException( |
2062 | 2067 | std::string("Error - DebugString: ") + |
|
0 commit comments