diff options
Diffstat (limited to 'src/armnn/test')
-rw-r--r-- | src/armnn/test/UtilsTests.cpp | 14 |
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() |