From 51403b5b8ae92782e2ed467cae4c0783faa5a22b Mon Sep 17 00:00:00 2001
From: Georgios Pinitas <georgios.pinitas@arm.com>
Date: Tue, 18 Jun 2019 14:10:48 +0100
Subject: 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 | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

(limited to 'src/core/CL/kernels/CLL2NormalizeLayerKernel.cpp')

diff --git a/src/core/CL/kernels/CLL2NormalizeLayerKernel.cpp b/src/core/CL/kernels/CLL2NormalizeLayerKernel.cpp
index 00af590104..288e1e8e61 100644
--- a/src/core/CL/kernels/CLL2NormalizeLayerKernel.cpp
+++ b/src/core/CL/kernels/CLL2NormalizeLayerKernel.cpp
@@ -100,11 +100,11 @@ void CLL2NormalizeLayerKernel::configure(const ICLTensor *input, const ICLTensor
     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 @@ void CLL2NormalizeLayerKernel::configure(const ICLTensor *input, const ICLTensor
     // 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
-- 
cgit v1.2.1