From 9331aebb5b75a79a68d0d3d541aa70032d255294 Mon Sep 17 00:00:00 2001 From: Gian Marco Iodice Date: Thu, 10 Aug 2017 17:11:08 +0100 Subject: COMPMID-417 - Added method to set the Local Workgroup size hint Change-Id: I909e92ff5a0972817e6b403d05d254622f3265c7 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/83552 Reviewed-by: Moritz Pflanzer Tested-by: Kaizen --- arm_compute/core/CL/ICLKernel.h | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'arm_compute/core/CL/ICLKernel.h') diff --git a/arm_compute/core/CL/ICLKernel.h b/arm_compute/core/CL/ICLKernel.h index c5e95aad88..d96b9c026e 100644 --- a/arm_compute/core/CL/ICLKernel.h +++ b/arm_compute/core/CL/ICLKernel.h @@ -111,6 +111,17 @@ public: _kernel.setArg(idx++, value); } + /** Set the Local-Workgroup-Size hint + * + * @note This method should be called after the configuration of the kernel + * + * @param[in] lws_hint Local-Workgroup-Size to use + */ + void set_lws_hint(cl::NDRange &lws_hint) + { + _lws_hint = lws_hint; + } + /** Set the targeted GPU architecture * * @param[in] target The targeted GPU architecture -- cgit v1.2.1