From e6c3344f2a76b0724172200b1d5a443393045336 Mon Sep 17 00:00:00 2001 From: Sang-Hoon Park Date: Tue, 17 Sep 2019 08:59:09 +0100 Subject: COMPMID-2637 [CL] fix broadcast pixel-wise multiplication with 5D tensors Broadcast pixel-wise multiplication with 5D tensors is fixed by adding information whether a dimension has been broadcasted to compute correct start offset when adding 3D tensor argument. The testcase that failed is added to the validation test suite. Change-Id: I320876f507012c27b39daae1316f9b69138ed204 Signed-off-by: Sang-Hoon Park Reviewed-on: https://review.mlplatform.org/c/1994 Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins Reviewed-by: Georgios Pinitas --- src/core/CL/ICLKernel.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src') diff --git a/src/core/CL/ICLKernel.cpp b/src/core/CL/ICLKernel.cpp index 2d28a496c9..d81ad46b29 100644 --- a/src/core/CL/ICLKernel.cpp +++ b/src/core/CL/ICLKernel.cpp @@ -98,7 +98,7 @@ void ICLKernel::add_tensor_argument(unsigned &idx, const ICLTensor *tensor, cons for(unsigned int n = 0; n < info->num_dimensions(); ++n) { - offset_first_element += window[n].start() * strides[n]; + offset_first_element += (window.is_broadcasted(n) ? 0 : window[n].start()) * strides[n]; } unsigned int idx_start = idx; -- cgit v1.2.1