diff options
Diffstat (limited to 'arm_compute/core')
-rw-r--r-- | arm_compute/core/CL/ICLKernel.h | 11 |
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 |