diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-05-11 19:51:14 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-05-11 19:51:14 +0100 |
commit | 878e0f94c43a7fd4f51dc7b39293409f46832520 (patch) | |
tree | 8365e27ae40ee01a4dc7137273cf74919a3664df | |
parent | 1e5afee25d460be70260ef81abd246a14fb2be8c (diff) | |
download | armnn-878e0f94c43a7fd4f51dc7b39293409f46832520.tar.gz |
IVGCVSW-5818 Unload network before checking result and improve test
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ic29592f808352b8a240bdafa9f2f8a09a737acfd
-rw-r--r-- | src/backends/cl/test/ClImportTensorHandleTests.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/backends/cl/test/ClImportTensorHandleTests.cpp b/src/backends/cl/test/ClImportTensorHandleTests.cpp index 85ff35f0af..3c8bd678c4 100644 --- a/src/backends/cl/test/ClImportTensorHandleTests.cpp +++ b/src/backends/cl/test/ClImportTensorHandleTests.cpp @@ -167,6 +167,8 @@ BOOST_FIXTURE_TEST_CASE(ClImportEndToEnd, ClContextControlFixture) auto outputData = std::make_unique<uint8_t[]>(space); void* alignedOutputPtr = outputData.get(); BOOST_CHECK(std::align(alignment, totalBytes, alignedOutputPtr, space)); + auto* outputPtr = reinterpret_cast<float*>(alignedOutputPtr); + std::fill_n(outputPtr, numElements, -10.0f); InputTensors inputTensors { @@ -200,6 +202,8 @@ BOOST_FIXTURE_TEST_CASE(ClImportEndToEnd, ClContextControlFixture) found = dump.find("CopyMemGeneric"); BOOST_TEST(found == std::string::npos); + runtime->UnloadNetwork(netId); + // Check output is as expected // Validate result by checking that the output has no negative values auto* outputResult = reinterpret_cast<float*>(alignedOutputPtr); @@ -208,8 +212,6 @@ BOOST_FIXTURE_TEST_CASE(ClImportEndToEnd, ClContextControlFixture) { BOOST_TEST(outputResult[i] >= 0); } - - runtime->UnloadNetwork(netId); } BOOST_AUTO_TEST_SUITE_END() |