From 3974a0469aec0eb79ca2384fd92c28d0329ef418 Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Fri, 16 Nov 2018 14:34:23 +0000 Subject: COMPMID-1791 CLPriorBox clCreateKernel error Change-Id: I2c2250669829e399fdc2363f729dc5e68d8aac17 --- src/core/CL/cl_kernels/prior_box_layer.cl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/core/CL/cl_kernels/prior_box_layer.cl') 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; -- cgit v1.2.1