From 14c9389923a2e59cc1e441615c1ff3cc150e9594 Mon Sep 17 00:00:00 2001 From: George Wort Date: Fri, 18 Jan 2019 15:51:25 +0000 Subject: COMPMID-1710: Fix address space issue Change-Id: I72ca4245b4732f304aa12e58cb0093773917cdf1 Reviewed-on: https://review.mlplatform.org/539 Tested-by: Arm Jenkins Reviewed-by: Giuseppe Rossini --- src/core/CL/cl_kernels/roi_align_layer.cl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/core') diff --git a/src/core/CL/cl_kernels/roi_align_layer.cl b/src/core/CL/cl_kernels/roi_align_layer.cl index a956860be2..430369b172 100644 --- a/src/core/CL/cl_kernels/roi_align_layer.cl +++ b/src/core/CL/cl_kernels/roi_align_layer.cl @@ -182,9 +182,9 @@ __kernel void roi_align_layer( for(int pz = 0; pz < MAX_DIM_Z; ++pz) { #if defined(NHWC) - DATA_TYPE *_output_ptr = (__global DATA_TYPE *)tensor3D_offset(&output, pz, px, py); + __global DATA_TYPE *_output_ptr = (__global DATA_TYPE *)tensor3D_offset(&output, pz, px, py); #else // !defined(NHWC) - DATA_TYPE *_output_ptr = (__global DATA_TYPE *)tensor3D_offset(&output, px, py, pz); + __global DATA_TYPE *_output_ptr = (__global DATA_TYPE *)tensor3D_offset(&output, px, py, pz); #endif // defined(NHWC) *_output_ptr = (__global DATA_TYPE)roi_align_1x1(&input, region_start.x, -- cgit v1.2.1