diff options
Diffstat (limited to 'src/backends/neon/test/NeonLayerSupportTests.cpp')
-rw-r--r-- | src/backends/neon/test/NeonLayerSupportTests.cpp | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/backends/neon/test/NeonLayerSupportTests.cpp b/src/backends/neon/test/NeonLayerSupportTests.cpp index 893f11ebf3..56955432b6 100644 --- a/src/backends/neon/test/NeonLayerSupportTests.cpp +++ b/src/backends/neon/test/NeonLayerSupportTests.cpp @@ -3,6 +3,8 @@ // SPDX-License-Identifier: MIT // +#include "NeonWorkloadFactoryHelper.hpp" + #include <layers/ConvertFp16ToFp32Layer.hpp> #include <layers/ConvertFp32ToFp16Layer.hpp> #include <test/TensorHelpers.hpp> @@ -20,19 +22,19 @@ BOOST_AUTO_TEST_SUITE(NeonLayerSupport) BOOST_AUTO_TEST_CASE(IsLayerSupportedFloat16Neon) { - armnn::NeonWorkloadFactory factory; + armnn::NeonWorkloadFactory factory = NeonWorkloadFactoryHelper::GetFactory(); IsLayerSupportedTests<armnn::NeonWorkloadFactory, armnn::DataType::Float16>(&factory); } BOOST_AUTO_TEST_CASE(IsLayerSupportedFloat32Neon) { - armnn::NeonWorkloadFactory factory; + armnn::NeonWorkloadFactory factory = NeonWorkloadFactoryHelper::GetFactory(); IsLayerSupportedTests<armnn::NeonWorkloadFactory, armnn::DataType::Float32>(&factory); } BOOST_AUTO_TEST_CASE(IsLayerSupportedUint8Neon) { - armnn::NeonWorkloadFactory factory; + armnn::NeonWorkloadFactory factory = NeonWorkloadFactoryHelper::GetFactory(); IsLayerSupportedTests<armnn::NeonWorkloadFactory, armnn::DataType::QuantisedAsymm8>(&factory); } |