From 8f6fe24102062883ef41be7b3a4aaf5e840e5bd9 Mon Sep 17 00:00:00 2001 From: Dongju Chae Date: Mon, 31 May 2021 15:47:53 +0900 Subject: [PATCH] [Svace] Fix memory leak This patch fixes memory leak. Signed-off-by: Dongju Chae --- tests/unittests/ne_core_data_test.cc | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/unittests/ne_core_data_test.cc b/tests/unittests/ne_core_data_test.cc index 116695f..f42f4b4 100644 --- a/tests/unittests/ne_core_data_test.cc +++ b/tests/unittests/ne_core_data_test.cc @@ -400,10 +400,12 @@ TEST (ne_core_data_test, dummy_quantize) { uint32_t data_dims[] = {1, 1, 1, 1024}; void *src_data = malloc (4096); - void *dst_data = malloc (4096); - ASSERT_NE (src_data, nullptr); - ASSERT_NE (dst_data, nullptr); + void *dst_data = malloc (4096); + if (dst_data == nullptr) { + free (src_data); + ASSERT_NE (dst_data, nullptr); + } /* std to npu */ std::unique_ptr converter (new DataConverter (true)); -- 2.7.4