IVGCVSW-5036 Do not allocate memory when import is enabled
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ideaae5280702aae6c73f3b4e4cee9f71a8386fda
diff --git a/src/backends/backendsCommon/test/EndToEndTestImpl.hpp b/src/backends/backendsCommon/test/EndToEndTestImpl.hpp
index a4d0d50..0d16bcd 100644
--- a/src/backends/backendsCommon/test/EndToEndTestImpl.hpp
+++ b/src/backends/backendsCommon/test/EndToEndTestImpl.hpp
@@ -719,6 +719,11 @@
std::vector<float> outputData0(4);
std::vector<float> outputData1(4);
+ std::vector<float> expectedOutput
+ {
+ 1.0f, 4.0f, 9.0f, 16.0f
+ };
+
InputTensors inputTensors
{
{0,armnn::ConstTensor(runtime->GetInputTensorInfo(netId, 0), inputData.data())},
@@ -764,6 +769,12 @@
// Contains CopyMemGeneric
found = dump.find("CopyMemGeneric");
BOOST_TEST(found != std::string::npos);
+
+ // Check that the outputs are correct
+ BOOST_CHECK_EQUAL_COLLECTIONS(outputData0.begin(), outputData0.end(),
+ expectedOutput.begin(), expectedOutput.end());
+ BOOST_CHECK_EQUAL_COLLECTIONS(outputData1.begin(), outputData1.end(),
+ expectedOutput.begin(), expectedOutput.end());
}
inline void StridedSliceInvalidSliceEndToEndTest(std::vector<BackendId> backends)