aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/CL/ICLKernel.h
diff options
context:
space:
mode:
authorGian Marco Iodice <gianmarco.iodice@arm.com>2017-08-10 17:11:08 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:35:24 +0000
commit9331aebb5b75a79a68d0d3d541aa70032d255294 (patch)
tree65dc3ba31dc51346b8fd0b0a1f9793e930f30fd6 /arm_compute/core/CL/ICLKernel.h
parent40ff03b07547ed1e145e956141c2c355e87515ee (diff)
downloadComputeLibrary-9331aebb5b75a79a68d0d3d541aa70032d255294.tar.gz
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 <moritz.pflanzer@arm.com> Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Diffstat (limited to 'arm_compute/core/CL/ICLKernel.h')
-rw-r--r--arm_compute/core/CL/ICLKernel.h11
1 files changed, 11 insertions, 0 deletions
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