commit c1f6b09cc22b6e2de3d9eb88aec1778d0308a2b3
author Teresa Charlin <teresa.charlinreyes@arm.com> Mon May 11 16:10:38 2020 +0100
committer TeresaARM <teresa.charlinreyes@arm.com> Wed May 13 18:34:12 2020 +0000
tree 14a70415b6d17b42246798a293dfdd13cbe4c123
parent 0c32ccfcd0b770cdd4eeb9d778b7f72a233e229a

IVGCVSW-4753 Refactor CL Softmax workload generalizing for different datatype

    * Change ComputeSoftmaxAclAxis to work with int and uint axis


Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ibbfa9ec7e2f0416e6885673212a767419c871cca

src/backends/cl/ClWorkloadFactory.cpp

diff --git a/src/backends/cl/ClWorkloadFactory.cpp b/src/backends/cl/ClWorkloadFactory.cpp
index b0d2fdf..f584272 100644
--- a/src/backends/cl/ClWorkloadFactory.cpp
+++ b/src/backends/cl/ClWorkloadFactory.cpp
@@ -490,8 +490,7 @@
 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateSoftmax(const SoftmaxQueueDescriptor& descriptor,
                                                             const WorkloadInfo& info) const
 {
-    return MakeWorkload<ClSoftmaxFloatWorkload, ClSoftmaxUint8Workload>(descriptor, info,
-                                                                        m_MemoryManager->GetIntraLayerManager());
+    return std::make_unique<ClSoftmaxWorkload>(descriptor, info, m_MemoryManager->GetIntraLayerManager());
 }
 
 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateSpaceToBatchNd(const SpaceToBatchNdQueueDescriptor& descriptor,