diff options
Diffstat (limited to 'src/armnn/test/OptimizerTests.cpp')
| -rw-r--r-- | src/armnn/test/OptimizerTests.cpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/armnn/test/OptimizerTests.cpp b/src/armnn/test/OptimizerTests.cpp index fcfff1a807..7fe69a9380 100644 --- a/src/armnn/test/OptimizerTests.cpp +++ b/src/armnn/test/OptimizerTests.cpp @@ -615,11 +615,15 @@ public: BOOST_AUTO_TEST_CASE(BackendCapabilityTest) { BackendId backendId = "MockBackend"; + + armnn::BackendOptions::BackendOption nonConstWeights{"NonConstWeights", true}; + // MockBackend does not support the NonConstWeights capability - BOOST_CHECK(!armnn::IsCapabilitySupported(backendId, armnn::BackendCapability::NonConstWeights)); + BOOST_CHECK(!armnn::HasCapability(nonConstWeights, backendId)); + BOOST_CHECK(!armnn::HasCapability("NonConstWeights", backendId)); // MockBackend does not support the AsyncExecution capability - BOOST_CHECK(!armnn::IsCapabilitySupported(backendId, armnn::BackendCapability::AsyncExecution)); + BOOST_CHECK(!armnn::GetCapability("AsyncExecution", backendId).has_value()); } BOOST_AUTO_TEST_CASE(BackendHintTest) |