commit 51403b5b8ae92782e2ed467cae4c0783faa5a22b
author Georgios Pinitas <georgios.pinitas@arm.com> Tue Jun 18 14:10:48 2019 +0100
committer Georgios Pinitas <georgios.pinitas@arm.com> Tue Jun 18 15:43:18 2019 +0000
tree 1a331a215df09a44056f4137e4fae7143b3477e4
parent 917539225a4283c869b3cb2c82bdc94b5d2c7756

COMPMID-2336: Fix argument type for epsilon in CLL2NormalizeLayer.

Change-Id: Iee3ef6d022f497df0e2e6227e06462762042b394
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1367
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>

src/core/CL/kernels/CLL2NormalizeLayerKernel.cpp

diff --git a/src/core/CL/kernels/CLL2NormalizeLayerKernel.cpp b/src/core/CL/kernels/CLL2NormalizeLayerKernel.cpp
index 00af590..288e1e8 100644
--- a/src/core/CL/kernels/CLL2NormalizeLayerKernel.cpp
+++ b/src/core/CL/kernels/CLL2NormalizeLayerKernel.cpp
@@ -100,11 +100,11 @@
     ARM_COMPUTE_ERROR_ON_NULLPTR(input, sum, output);
     ARM_COMPUTE_ERROR_THROW_ON(validate_arguments(input->info(), sum->info(), output->info(), axis, epsilon));
 
-    _input   = input;
-    _sum     = sum;
-    _output  = output;
-    _actual_axis    = wrap_around(axis, max_input_tensor_dim);
-    _epsilon = epsilon;
+    _input       = input;
+    _sum         = sum;
+    _output      = output;
+    _actual_axis = wrap_around(axis, max_input_tensor_dim);
+    _epsilon     = epsilon;
 
     const unsigned int num_elems_processed_per_iteration = 16;
 
@@ -138,11 +138,11 @@
     // Set epsilon argument
     if(input->info()->data_type() == DataType::F32)
     {
-        _kernel.setArg<cl_uint>(idx, _epsilon);
+        _kernel.setArg<cl_float>(idx, _epsilon);
     }
     else
     {
-        _kernel.setArg<cl_ushort>(idx, _epsilon);
+        _kernel.setArg<cl_half>(idx, _epsilon);
     }
 
     // Configure kernel window