From: Inki Dae <inki.dae@samsung.com>
Date: Thu, 5 Mar 2020 01:25:21 +0000 (+0900)
Subject: test: Release tensor buffers after inference
X-Git-Tag: submit/tizen/20200423.063253~39
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=1c9b90859177700c6ca9a0078b6dd72070ffb5ee;p=platform%2Fcore%2Fmultimedia%2Finference-engine-interface.git

test: Release tensor buffers after inference

Change-Id: I27cfdd2ae2a9d1ee97c60c3211e309e26334399e
Signed-off-by: Inki Dae <inki.dae@samsung.com>
---

diff --git a/test/src/inference_engine_test.cpp b/test/src/inference_engine_test.cpp
index 27125d1..96f6d33 100644
--- a/test/src/inference_engine_test.cpp
+++ b/test/src/inference_engine_test.cpp
@@ -182,6 +182,33 @@ int PrepareTensorBuffers(InferenceEngineCommon *engine, std::vector<inference_en
     return INFERENCE_ENGINE_ERROR_NONE;
 }
 
+void CleanupTensorBuffers(std::vector<inference_engine_tensor_buffer> &inputs, std::vector<inference_engine_tensor_buffer> &outputs)
+{
+	if (!inputs.empty()) {
+		std::vector<inference_engine_tensor_buffer>::iterator iter;
+		for (iter = inputs.begin(); iter != inputs.end(); iter++) {
+			inference_engine_tensor_buffer tensor_buffer = *iter;
+			if (tensor_buffer.data_type == TENSOR_DATA_TYPE_FLOAT32)
+				delete[] (float *)tensor_buffer.buffer;
+			else
+				delete[] (unsigned char *)tensor_buffer.buffer;
+		}
+		std::vector<inference_engine_tensor_buffer>().swap(inputs);
+	}
+
+	if (!outputs.empty()) {
+		std::vector<inference_engine_tensor_buffer>::iterator iter;
+		for (iter = outputs.begin(); iter != outputs.end(); iter++) {
+			inference_engine_tensor_buffer tensor_buffer = *iter;
+			if (tensor_buffer.data_type == TENSOR_DATA_TYPE_FLOAT32)
+				delete[] (float *)tensor_buffer.buffer;
+			else
+				delete[] (unsigned char *)tensor_buffer.buffer;
+		}
+		std::vector<inference_engine_tensor_buffer>().swap(outputs);
+	}
+}
+
 void CopyFileToMemory(const char *file_name, inference_engine_tensor_buffer &buffer, unsigned int size)
 {
     int fd = open(file_name, O_RDONLY);
@@ -497,6 +524,8 @@ TEST_P(InferenceEngineCommonTest_3, Inference)
         break;
     }
 
+    CleanupTensorBuffers(inputs, outputs);
+
     engine->UnbindBackend();
 
     delete engine;