Block memory release in case of tensorflow.
authorHyunsoo Park <hance.park@samsung.com>
Thu, 2 Apr 2020 09:13:49 +0000 (18:13 +0900)
committerInki Dae <inki.dae@samsung.com>
Tue, 14 Apr 2020 00:42:53 +0000 (09:42 +0900)
Change-Id: Ibb2e79822e141e3ed7216156a38ab26fcd720e02
Signed-off-by: Hyunsoo Park <hance.park@samsung.com>
test/src/inference_engine_test.cpp

index dce61cba426f80ff36803d108cddf20a1c51a805..dbcd54a347405d45cf31b2e6dbff5f94c5dfa2cf 100644 (file)
@@ -323,7 +323,9 @@ void FillOutputResult(InferenceEngineCommon *engine, std::vector<inference_engin
 
                        // replace original buffer with new one, and release origin one.
                        outputs[i].buffer = new_buf;
-                       delete[] ori_buf;
+                       if (!outputs[i].owner_is_backend) {
+                       delete[] ori_buf;
+                       }
                }
 
                outputData.data.push_back((void *)outputs[i].buffer);
@@ -648,4 +650,4 @@ INSTANTIATE_TEST_CASE_P(Prefix, InferenceEngineCommonTest_3,
                                                           76, 146, 170, 193, 216, 146, 123,  99, 287, 381, 451, 287, 381, 475 })
                        /* TODO */
                )
-);
\ No newline at end of file
+);