From 7f466e438f1ad28fd016965d801641b0eabc037b Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=9D=B4=EC=83=81=EA=B7=9C/On-Device=20Lab=28SR=29/Princip?= =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Fri, 16 Aug 2019 13:28:12 +0900 Subject: [PATCH] [nnpackage_run] Fix wrong allocation size for output (#6640) outputs[i] is vector. sizeof(float) should not be multiplied. Signed-off-by: Sanggyu Lee --- tests/tools/nnpackage_run/src/nnpackage_run.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/tools/nnpackage_run/src/nnpackage_run.cc b/tests/tools/nnpackage_run/src/nnpackage_run.cc index 862ab90..9a6383a 100644 --- a/tests/tools/nnpackage_run/src/nnpackage_run.cc +++ b/tests/tools/nnpackage_run/src/nnpackage_run.cc @@ -111,7 +111,7 @@ int main(const int argc, char **argv) for (uint32_t j = 0; j < ti.rank; ++j) output_num_elements *= ti.dims[j]; - outputs[i].resize(sizeof(float) * output_num_elements); + outputs[i].resize(output_num_elements); NNPR_ENSURE_STATUS(nnfw_set_output(session, i, NNFW_TYPE_TENSOR_FLOAT32, outputs[i].data(), sizeof(float) * output_num_elements)); } -- 2.7.4