IVGCVSW-6558 Fix the failing OptimisedNetworkTests
* With the new Memory Management updates input tensors should now match the backend of the layer they are attached to. Updated the test to reflect this.
Signed-off-by: David Monahan <David.Monahan@arm.com>
Change-Id: I4e7fbd3f2f40d349c39cdf0d7bfef1262144181b
diff --git a/src/backends/backendsCommon/test/OptimizedNetworkTests.cpp b/src/backends/backendsCommon/test/OptimizedNetworkTests.cpp
index 6eecaab..4b932c7 100644
--- a/src/backends/backendsCommon/test/OptimizedNetworkTests.cpp
+++ b/src/backends/backendsCommon/test/OptimizedNetworkTests.cpp
@@ -337,7 +337,11 @@
// the other layers are supported by CpuRef.
// If neither NEON, nor CL is enabled, all layers are supported by CpuRef.
#if defined(ARMCOMPUTENEON_ENABLED)
- if (layer->GetType() == armnn::LayerType::Output)
+ if (layer->GetType() == armnn::LayerType::Input)
+ {
+ CHECK(layer->GetBackendId() == armnn::Compute::CpuRef);
+ }
+ else if (layer->GetType() == armnn::LayerType::Output)
{
CHECK(layer->GetBackendId() == armnn::Compute::CpuAcc);
}
@@ -346,7 +350,11 @@
CHECK(layer->GetBackendId() == armnn::Compute::CpuRef);
}
#elif defined(ARMCOMPUTECL_ENABLED)
- if (layer->GetType() == armnn::LayerType::Input || layer->GetType() == armnn::LayerType::Output)
+ if (layer->GetType() == armnn::LayerType::Input)
+ {
+ CHECK(layer->GetBackendId() == armnn::Compute::CpuRef);
+ }
+ else if (layer->GetType() == armnn::LayerType::Output)
{
CHECK(layer->GetBackendId() == armnn::Compute::GpuAcc);
}