diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/backends/cl/ClWorkloadFactory.cpp | 3 | ||||
-rw-r--r-- | src/backends/cl/workloads/ClAdditionWorkload.cpp | 13 | ||||
-rw-r--r-- | src/backends/cl/workloads/ClAdditionWorkload.hpp | 3 | ||||
-rw-r--r-- | src/backends/test/CreateWorkloadCl.cpp | 4 |
4 files changed, 8 insertions, 15 deletions
diff --git a/src/backends/cl/ClWorkloadFactory.cpp b/src/backends/cl/ClWorkloadFactory.cpp index c00d0ba5f0..b4f4a72e9d 100644 --- a/src/backends/cl/ClWorkloadFactory.cpp +++ b/src/backends/cl/ClWorkloadFactory.cpp @@ -162,8 +162,7 @@ std::unique_ptr<armnn::IWorkload> ClWorkloadFactory::CreateNormalization(const N std::unique_ptr<armnn::IWorkload> ClWorkloadFactory::CreateAddition(const AdditionQueueDescriptor& descriptor, const WorkloadInfo& info) const { - return MakeWorkload<ClAdditionWorkload<armnn::DataType::Float16, armnn::DataType::Float32>, - ClAdditionWorkload<armnn::DataType::QuantisedAsymm8>>(descriptor, info); + return std::make_unique<ClAdditionWorkload>(descriptor, info); } std::unique_ptr<armnn::IWorkload> ClWorkloadFactory::CreateMultiplication( diff --git a/src/backends/cl/workloads/ClAdditionWorkload.cpp b/src/backends/cl/workloads/ClAdditionWorkload.cpp index ec0dd30275..c9ac958402 100644 --- a/src/backends/cl/workloads/ClAdditionWorkload.cpp +++ b/src/backends/cl/workloads/ClAdditionWorkload.cpp @@ -17,10 +17,9 @@ using namespace armcomputetensorutils; static constexpr arm_compute::ConvertPolicy g_AclConvertPolicy = arm_compute::ConvertPolicy::SATURATE; -template <armnn::DataType... T> -ClAdditionWorkload<T...>::ClAdditionWorkload(const AdditionQueueDescriptor& descriptor, - const WorkloadInfo& info) - : TypedWorkload<AdditionQueueDescriptor, T...>(descriptor, info) +ClAdditionWorkload::ClAdditionWorkload(const AdditionQueueDescriptor& descriptor, + const WorkloadInfo& info) + : BaseWorkload<AdditionQueueDescriptor>(descriptor, info) { this->m_Data.ValidateInputsOutputs("ClAdditionWorkload", 2, 1); @@ -30,8 +29,7 @@ ClAdditionWorkload<T...>::ClAdditionWorkload(const AdditionQueueDescriptor& desc m_Layer.configure(&input0, &input1, &output, g_AclConvertPolicy); } -template <armnn::DataType... T> -void ClAdditionWorkload<T...>::Execute() const +void ClAdditionWorkload::Execute() const { ARMNN_SCOPED_PROFILING_EVENT_CL("ClAdditionWorkload_Execute"); m_Layer.run(); @@ -54,6 +52,3 @@ arm_compute::Status ClAdditionValidate(const TensorInfo& input0, } } //namespace armnn - -template class armnn::ClAdditionWorkload<armnn::DataType::Float16, armnn::DataType::Float32>; -template class armnn::ClAdditionWorkload<armnn::DataType::QuantisedAsymm8>; diff --git a/src/backends/cl/workloads/ClAdditionWorkload.hpp b/src/backends/cl/workloads/ClAdditionWorkload.hpp index c5e6affac8..d165bd90f7 100644 --- a/src/backends/cl/workloads/ClAdditionWorkload.hpp +++ b/src/backends/cl/workloads/ClAdditionWorkload.hpp @@ -12,8 +12,7 @@ namespace armnn { -template <armnn::DataType... dataTypes> -class ClAdditionWorkload : public TypedWorkload<AdditionQueueDescriptor, dataTypes...> +class ClAdditionWorkload : public BaseWorkload<AdditionQueueDescriptor> { public: ClAdditionWorkload(const AdditionQueueDescriptor& descriptor, const WorkloadInfo& info); diff --git a/src/backends/test/CreateWorkloadCl.cpp b/src/backends/test/CreateWorkloadCl.cpp index 411f72baee..d56bad2bb9 100644 --- a/src/backends/test/CreateWorkloadCl.cpp +++ b/src/backends/test/CreateWorkloadCl.cpp @@ -69,7 +69,7 @@ static void ClCreateArithmethicWorkloadTest() BOOST_AUTO_TEST_CASE(CreateAdditionFloatWorkload) { - ClCreateArithmethicWorkloadTest<ClAdditionWorkload<armnn::DataType::Float16, armnn::DataType::Float32>, + ClCreateArithmethicWorkloadTest<ClAdditionWorkload, AdditionQueueDescriptor, AdditionLayer, armnn::DataType::Float32>(); @@ -77,7 +77,7 @@ BOOST_AUTO_TEST_CASE(CreateAdditionFloatWorkload) BOOST_AUTO_TEST_CASE(CreateAdditionFloat16Workload) { - ClCreateArithmethicWorkloadTest<ClAdditionWorkload<armnn::DataType::Float16, armnn::DataType::Float32>, + ClCreateArithmethicWorkloadTest<ClAdditionWorkload, AdditionQueueDescriptor, AdditionLayer, armnn::DataType::Float16>(); |