From 8e74bc51481aee432f1667769ec96e0d7ce65b4c Mon Sep 17 00:00:00 2001 From: Comma Device Date: Tue, 7 Jun 2022 15:20:19 -0700 Subject: [PATCH] ; --- selfdrive/modeld/runners/snpemodel.cc | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/selfdrive/modeld/runners/snpemodel.cc b/selfdrive/modeld/runners/snpemodel.cc index cdaf02b0ad..482169ebc6 100644 --- a/selfdrive/modeld/runners/snpemodel.cc +++ b/selfdrive/modeld/runners/snpemodel.cc @@ -79,7 +79,7 @@ SNPEModel::SNPEModel(const char *path, float *loutput, size_t loutput_size, int const auto &inputDims_opt = snpe->getInputDimensions(input_tensor_name); const zdl::DlSystem::TensorShape& bufferShape = *inputDims_opt; std::vector strides(bufferShape.rank()); - strides[strides.size() - 1] = size_of_input + strides[strides.size() - 1] = size_of_input; size_t product = 1; for (size_t i = 0; i < bufferShape.rank(); i++) product *= bufferShape[i]; size_t stride = strides[strides.size() - 1]; @@ -89,9 +89,9 @@ SNPEModel::SNPEModel(const char *path, float *loutput, size_t loutput_size, int } printf("input product is %lu\n", product); inputBuffer = ubFactory.createUserBuffer(NULL, - product*size_of_input, - strides, - use_tf8 ? &userBufferEncodingTf8 : &userBufferEncodingFloat); + product*size_of_input, + strides, + use_tf8 ? (zdl::DlSystem::UserBufferEncoding*)&userBufferEncodingTf8 : (zdl::DlSystem::UserBufferEncoding*)&userBufferEncodingFloat); inputMap.add(input_tensor_name, inputBuffer.get()); }