diff options
author | George Wort <george.wort@arm.com> | 2019-01-18 15:51:25 +0000 |
---|---|---|
committer | George Wort <george.wort@arm.com> | 2019-01-18 16:48:12 +0000 |
commit | 14c9389923a2e59cc1e441615c1ff3cc150e9594 (patch) | |
tree | 0a8d7e893499c52673487273fc930d83b2835f0e /src/core/CL/cl_kernels/roi_align_layer.cl | |
parent | 7cd26d4a1b14bc4bf7c61496803416ab3d84791f (diff) | |
download | ComputeLibrary-14c9389923a2e59cc1e441615c1ff3cc150e9594.tar.gz |
COMPMID-1710: Fix address space issue
Change-Id: I72ca4245b4732f304aa12e58cb0093773917cdf1
Reviewed-on: https://review.mlplatform.org/539
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Giuseppe Rossini <giuseppe.rossini@arm.com>
Diffstat (limited to 'src/core/CL/cl_kernels/roi_align_layer.cl')
-rw-r--r-- | src/core/CL/cl_kernels/roi_align_layer.cl | 4 |
1 files changed, 2 insertions, 2 deletions
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, |