aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMichalis Spyrou <michalis.spyrou@arm.com>2018-11-16 14:34:23 +0000
committerMichalis Spyrou <michalis.spyrou@arm.com>2018-11-16 14:34:23 +0000
commit3974a0469aec0eb79ca2384fd92c28d0329ef418 (patch)
treeddea501a734ac91fb86b64e1932cf6c8883a5858
parent3f9d4d7c7242f50330dca4ff104a76bc88804903 (diff)
downloadComputeLibrary-3974a0469aec0eb79ca2384fd92c28d0329ef418.tar.gz
COMPMID-1791 CLPriorBox clCreateKernel error
Change-Id: I2c2250669829e399fdc2363f729dc5e68d8aac17
-rw-r--r--src/core/CL/cl_kernels/prior_box_layer.cl4
1 files changed, 2 insertions, 2 deletions
diff --git a/src/core/CL/cl_kernels/prior_box_layer.cl b/src/core/CL/cl_kernels/prior_box_layer.cl
index 2a0329d239..be072ecb1d 100644
--- a/src/core/CL/cl_kernels/prior_box_layer.cl
+++ b/src/core/CL/cl_kernels/prior_box_layer.cl
@@ -72,7 +72,7 @@ inline void calculate_xy_min_max_nchw(Image *out, int idx, float center_x, float
*
* @return The updated index
*/
-inline int calculate_min_nchw(Image *out, float *max, float *aspect_ratios, int max_size, int aspect_ratios_size, float min_size, int min_idx, int idx)
+inline int calculate_min_nchw(Image *out, __global float *max, __global float *aspect_ratios, int max_size, int aspect_ratios_size, float min_size, int min_idx, int idx)
{
const float center_x = ((float)(get_global_id(0) % LAYER_WIDTH) + OFFSET) * STEP_X;
const float center_y = ((float)(get_global_id(0) / LAYER_WIDTH) + OFFSET) * STEP_Y;
@@ -153,7 +153,7 @@ inline void calculate_xy_min_max_nhwc(Tensor3D *out, int idx, float center_x, fl
*
* @return The updated index
*/
-inline int calculate_min_nhwc(Image *out, float *max, float *aspect_ratios, int max_size, int aspect_ratios_size, float min_size, int min_idx, int idx)
+inline int calculate_min_nhwc(Tensor3D *out, __global float *max, __global float *aspect_ratios, int max_size, int aspect_ratios_size, float min_size, int min_idx, int idx)
{
const float center_x = ((float)(get_global_id(1) % LAYER_WIDTH) + OFFSET) * STEP_X;
const float center_y = ((float)(get_global_id(1) / LAYER_WIDTH) + OFFSET) * STEP_Y;