From b9af86ea42568ade799ee5529137e4756977b6c6 Mon Sep 17 00:00:00 2001 From: Finn Williams Date: Wed, 26 May 2021 18:38:12 +0100 Subject: IVGCVSW-5855 Refactor the reporting of capabilities from backends Signed-off-by: Finn Williams Change-Id: I05fc331a8e91bdcb6b8a2f32cfb555060fc5d797 --- src/armnn/test/OptimizerTests.cpp | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'src/armnn/test/OptimizerTests.cpp') 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) -- cgit v1.2.1