From 8468371b3e2ec42ee0b9b670d45d99eb1015574b Mon Sep 17 00:00:00 2001 From: Ramy Elgammal Date: Fri, 16 Dec 2022 13:39:33 +0000 Subject: Make CLReshape kernel window based on dst instead of src Signed-off-by: Ramy Elgammal Partially-Resolves: COMPMID-5522 Change-Id: I1d90003079c3f24d081cc49f7b110eda753f6995 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/8838 Reviewed-by: Gian Marco Iodice Benchmark: Arm Jenkins Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins --- src/gpu/cl/kernels/ClReshapeKernel.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/gpu') diff --git a/src/gpu/cl/kernels/ClReshapeKernel.cpp b/src/gpu/cl/kernels/ClReshapeKernel.cpp index 246bd9c838..121bb33edf 100644 --- a/src/gpu/cl/kernels/ClReshapeKernel.cpp +++ b/src/gpu/cl/kernels/ClReshapeKernel.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2021 Arm Limited. + * Copyright (c) 2017-2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -98,7 +98,7 @@ void ClReshapeKernel::configure(const CLCompileContext &compile_context, const I _kernel.setArg(idx++, dst_shape); // Configure kernel window - Window win = calculate_max_window(*src); + Window win = calculate_max_window(*dst); ICLKernel::configure_internal(win); ARM_COMPUTE_ERROR_ON(has_padding_changed(padding_info)); -- cgit v1.2.1