diff options
Diffstat (limited to 'src/backends/cl/test/ClLayerSupportTests.cpp')
| -rw-r--r-- | src/backends/cl/test/ClLayerSupportTests.cpp | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/backends/cl/test/ClLayerSupportTests.cpp b/src/backends/cl/test/ClLayerSupportTests.cpp index 0019afed6b..2218d821ef 100644 --- a/src/backends/cl/test/ClLayerSupportTests.cpp +++ b/src/backends/cl/test/ClLayerSupportTests.cpp @@ -3,6 +3,8 @@ // SPDX-License-Identifier: MIT // +#include "ClWorkloadFactoryHelper.hpp" + #include <layers/ConvertFp16ToFp32Layer.hpp> #include <layers/ConvertFp32ToFp16Layer.hpp> #include <test/TensorHelpers.hpp> @@ -21,19 +23,19 @@ BOOST_AUTO_TEST_SUITE(ClLayerSupport) BOOST_FIXTURE_TEST_CASE(IsLayerSupportedFloat16Cl, ClContextControlFixture) { - armnn::ClWorkloadFactory factory; + armnn::ClWorkloadFactory factory = ClWorkloadFactoryHelper::GetFactory(); IsLayerSupportedTests<armnn::ClWorkloadFactory, armnn::DataType::Float16>(&factory); } BOOST_FIXTURE_TEST_CASE(IsLayerSupportedFloat32Cl, ClContextControlFixture) { - armnn::ClWorkloadFactory factory; + armnn::ClWorkloadFactory factory = ClWorkloadFactoryHelper::GetFactory(); IsLayerSupportedTests<armnn::ClWorkloadFactory, armnn::DataType::Float32>(&factory); } BOOST_FIXTURE_TEST_CASE(IsLayerSupportedUint8Cl, ClContextControlFixture) { - armnn::ClWorkloadFactory factory; + armnn::ClWorkloadFactory factory = ClWorkloadFactoryHelper::GetFactory(); IsLayerSupportedTests<armnn::ClWorkloadFactory, armnn::DataType::QuantisedAsymm8>(&factory); } |