commit 58307943d9a66552f8a425683061fd1aa2b599c4
author Michalis Spyrou <michalis.spyrou@arm.com> Fri Jan 17 16:36:46 2020 +0000
committer Georgios Pinitas <georgios.pinitas@arm.com> Mon Jan 20 16:38:51 2020 +0000
tree bef6d8cce8fc8d3d57c043daf6ece9cfb45f4cff
parent 0cb3da671b652641deff909a21f6e45a550452f1

COMPMID-2819 Unary Operation VTS/CTS failures on CL

Change-Id: I49231bb66101244d05d6eb35bc644bcc8693aa34
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/2602
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Georgios Pinitas <georgios.pinitas@arm.com>

src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp

diff --git a/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp b/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp
index c4ab504..543c8f3 100644
--- a/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp
+++ b/src/core/CL/kernels/CLElementWiseUnaryLayerKernel.cpp
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2018-2019 ARM Limited.
+ * Copyright (c) 2018-2020 ARM Limited.
  *
  * SPDX-License-Identifier: MIT
  *
@@ -121,14 +121,15 @@
     ARM_COMPUTE_ERROR_ON_UNCONFIGURED_KERNEL(this);
     ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICLKernel::window(), window);
 
-    Window collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimX);
+    Window collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ);
+    Window slice     = collapsed.first_slice_window_3D();
 
     do
     {
         unsigned int idx = 0;
-        add_1D_tensor_argument(idx, _input, collapsed);
-        add_1D_tensor_argument(idx, _output, collapsed);
-        enqueue(queue, *this, collapsed, lws_hint());
+        add_3D_tensor_argument(idx, _input, slice);
+        add_3D_tensor_argument(idx, _output, slice);
+        enqueue(queue, *this, slice, lws_hint());
     }
-    while(window.slide_window_slice_1D(collapsed));
+    while(collapsed.slide_window_slice_3D(slice));
 }