commit f24411ffc842970609a1fb6ba2f9527cfb681dbd
author giuros01 <giuseppe.rossini@arm.com> Thu May 16 11:47:13 2019 +0100
committer Giuseppe Rossini <giuseppe.rossini@arm.com> Tue May 21 11:14:06 2019 +0000
tree ce906a524722f37edf22ec257d0f62bb04cb3546
parent 2ef70cf813a20f12ad11900892c374edb9fa6ee4

COMPMID-2255: Implement ABS operator for CL

Change-Id: I8867953f8ae84a77f126aedd32b2f98c062905b1
Signed-off-by: giuros01 <giuseppe.rossini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1151
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>

src/core/CL/cl_kernels/elementwise_unary.cl

diff --git a/src/core/CL/cl_kernels/elementwise_unary.cl b/src/core/CL/cl_kernels/elementwise_unary.cl
index 14d6f5d..89bffed 100644
--- a/src/core/CL/cl_kernels/elementwise_unary.cl
+++ b/src/core/CL/cl_kernels/elementwise_unary.cl
@@ -26,14 +26,16 @@
 
 #if defined(DATA_TYPE) && defined(OPERATION)
 
-// Calculate Exponential
-#define exponential(input) exp(input)
+// Calculate exponential
+#define exp_op(input) exp(input)
 // Calculate reverse square root
-#define inverse_sqrt(input) rsqrt(input)
+#define rsqrt_op(input) rsqrt(input)
 // Calculate negative
-#define neg(input) (-input)
+#define neg_op(input) (-input)
 // Calculate sine
-#define sine(input) sin(input)
+#define sin_op(input) sin(input)
+// Calculate abs for floating point values
+#define fabs_op(input) fabs(input)
 
 /** Applies element wise unary operator in a tensor.
  *

src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp

diff --git a/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp b/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp
index b6484ec..15823b0 100644
--- a/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp
+++ b/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp
@@ -79,16 +79,19 @@
     switch(op)
     {
         case ElementWiseUnary::RSQRT:
-            build_opts.add_option("-DOPERATION=inverse_sqrt");
+            build_opts.add_option("-DOPERATION=rsqrt_op");
             break;
         case ElementWiseUnary::EXP:
-            build_opts.add_option("-DOPERATION=exponential");
+            build_opts.add_option("-DOPERATION=exp_op");
             break;
         case ElementWiseUnary::NEG:
-            build_opts.add_option("-DOPERATION=neg");
+            build_opts.add_option("-DOPERATION=neg_op");
             break;
         case ElementWiseUnary::SIN:
-            build_opts.add_option("-DOPERATION=sine");
+            build_opts.add_option("-DOPERATION=sin_op");
+            break;
+        case ElementWiseUnary::ABS:
+            build_opts.add_option("-DOPERATION=fabs_op");
             break;
         default:
             ARM_COMPUTE_ERROR("Not implemented");

src/runtime/CL/functions/CLElementWiseUnaryLayer.cpp

diff --git a/src/runtime/CL/functions/CLElementWiseUnaryLayer.cpp b/src/runtime/CL/functions/CLElementWiseUnaryLayer.cpp
index c3e472d..14e77ea 100644
--- a/src/runtime/CL/functions/CLElementWiseUnaryLayer.cpp
+++ b/src/runtime/CL/functions/CLElementWiseUnaryLayer.cpp
@@ -74,4 +74,15 @@
     return CLElementWiseUnaryLayerKernel::validate(input, output, ElementWiseUnary::SIN);
 }
 
+void CLAbsLayer::configure(const ICLTensor *input, ICLTensor *output)
+{
+    auto k = arm_compute::support::cpp14::make_unique<CLElementWiseUnaryLayerKernel>();
+    k->configure(input, output, ElementWiseUnary::ABS);
+    _kernel = std::move(k);
+}
+Status CLAbsLayer::validate(const ITensorInfo *input, const ITensorInfo *output)
+{
+    return CLElementWiseUnaryLayerKernel::validate(input, output, ElementWiseUnary::ABS);
+}
+
 } // namespace arm_compute