From: Dongju Chae <dongju.chae@samsung.com>
Date: Mon, 31 May 2021 06:47:53 +0000 (+0900)
Subject: [Svace] Fix memory leak
X-Git-Tag: accepted/tizen/unified/20220103.130045~143
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=8f6fe24102062883ef41be7b3a4aaf5e840e5bd9;p=platform%2Fadaptation%2Fnpu%2Ftrix-engine.git

[Svace] Fix memory leak

This patch fixes memory leak.

Signed-off-by: Dongju Chae <dongju.chae@samsung.com>
---

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<DataConverter> converter (new DataConverter (true));