[Test] Fix npu test leak

This patch fixes unittest_ne_core_npu for memory leak.
seg buffer is freed after used.

Signed-off-by: Jiho Chu <jiho.chu@samsung.com>

diff --git a/tests/unittests/ne_core_npu_test.cc b/tests/unittests/ne_core_npu_test.cc
index 264506e6836d5318a67be3eae1e5341d9ac14bf6..7600c04752a583a30e55f6e3f85334d95bed7b3a 100644 (file)
--- a/tests/unittests/ne_core_npu_test.cc
+++ b/tests/unittests/ne_core_npu_test.cc
@@ -345,6 +345,7 @@ TEST_F (NpuTest, run_inference_triv2) {
   EXPECT_GE (api->runInput (&input), 0);
   EXPECT_EQ (api->munmap (buf, size), 0);
   EXPECT_EQ (api->deregisterModel (model.id), 0);
+  EXPECT_EQ (api->dealloc (seg_dmabuf), 0);
   EXPECT_EQ (api->dealloc (model_dmabuf), 0);
   EXPECT_EQ (api->dealloc (buffer_dmabuf), 0);
 }
@@ -392,6 +393,7 @@ TEST_F (NpuTest, run_inference_async0_triv2) {
   EXPECT_GE (f_ret.get (), 0);
   EXPECT_EQ (api->munmap (buf, size), 0);
   EXPECT_EQ (api->deregisterModel (model.id), 0);
+  EXPECT_EQ (api->dealloc (seg_dmabuf), 0);
   EXPECT_EQ (api->dealloc (model_dmabuf), 0);
   EXPECT_EQ (api->dealloc (buffer_dmabuf), 0);
 }
@@ -444,6 +446,7 @@ TEST_F (NpuTest, run_inference_async1_triv2) {
   EXPECT_GE (api->runInput (&input), 0);
   EXPECT_EQ (api->munmap (buf, size), 0);
   EXPECT_EQ (api->deregisterModel (model.id), 0);
+  EXPECT_EQ (api->dealloc (seg_dmabuf), 0);
   EXPECT_EQ (api->dealloc (model.dbuf_fd), 0);
   EXPECT_EQ (api->dealloc (buffer_dmabuf), 0);
 }