aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/test/UtilsTests.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/test/UtilsTests.cpp')
-rw-r--r--src/armnn/test/UtilsTests.cpp14
1 files changed, 14 insertions, 0 deletions
diff --git a/src/armnn/test/UtilsTests.cpp b/src/armnn/test/UtilsTests.cpp
index fb078de32f..7776a2d3cf 100644
--- a/src/armnn/test/UtilsTests.cpp
+++ b/src/armnn/test/UtilsTests.cpp
@@ -5,6 +5,7 @@
#include <boost/test/unit_test.hpp>
+#include <armnn/BackendHelper.hpp>
#include <armnn/Utils.hpp>
#include <armnn/Types.hpp>
#include <armnn/TypesUtils.hpp>
@@ -266,4 +267,17 @@ BOOST_AUTO_TEST_CASE(PermuteQuantizationDim)
BOOST_CHECK(permuted.GetQuantizationDim().value() == 1U);
}
+#if defined(ARMNNREF_ENABLED)
+BOOST_AUTO_TEST_CASE(LayerSupportHandle)
+{
+ auto layerSupportObject = armnn::GetILayerSupportByBackendId("CpuRef");
+ armnn::TensorInfo input;
+ std::string reasonIfUnsupported;
+ // InputLayer always supported for CpuRef
+ BOOST_CHECK_EQUAL(layerSupportObject.IsInputSupported(input, reasonIfUnsupported), true);
+
+ BOOST_CHECK(layerSupportObject.IsBackendRegistered());
+}
+#endif
+
BOOST_AUTO_TEST_SUITE_END()