diff options
Diffstat (limited to 'src/armnn/test/UtilsTests.cpp')
-rw-r--r-- | src/armnn/test/UtilsTests.cpp | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/src/armnn/test/UtilsTests.cpp b/src/armnn/test/UtilsTests.cpp index 7776a2d3cf..f0198cb9d4 100644 --- a/src/armnn/test/UtilsTests.cpp +++ b/src/armnn/test/UtilsTests.cpp @@ -278,6 +278,18 @@ BOOST_AUTO_TEST_CASE(LayerSupportHandle) BOOST_CHECK(layerSupportObject.IsBackendRegistered()); } + +BOOST_AUTO_TEST_CASE(IsCapabilitySupported_CpuRef) +{ + BOOST_CHECK(armnn::IsCapabilitySupported(armnn::Compute::CpuRef, armnn::BackendCapability::NonConstWeights)); +} +#endif + +#if defined(ARMCOMPUTENEON_ENABLED) +BOOST_AUTO_TEST_CASE(IsCapabilitySupported_CpuAcc) +{ + BOOST_CHECK(!armnn::IsCapabilitySupported(armnn::Compute::CpuAcc, armnn::BackendCapability::NonConstWeights)); +} #endif BOOST_AUTO_TEST_SUITE_END() |