[nnpackage_run] Fix wrong allocation size for output (#6640)
author이상규/On-Device Lab(SR)/Principal Engineer/삼성전자 <sg5.lee@samsung.com>
Fri, 16 Aug 2019 04:28:12 +0000 (13:28 +0900)
committer이한종/On-Device Lab(SR)/Engineer/삼성전자 <hanjoung.lee@samsung.com>
Fri, 16 Aug 2019 04:28:12 +0000 (13:28 +0900)
outputs[i] is vector<float>. sizeof(float) should not be multiplied.

Signed-off-by: Sanggyu Lee <sg5.lee@samsung.com>
tests/tools/nnpackage_run/src/nnpackage_run.cc

index 862ab90..9a6383a 100644 (file)
@@ -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));
   }