From d24af8a3e8f7cbd38fd3142056241c0c9f63e46a Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Tue, 8 May 2018 17:23:52 +0100 Subject: COMPMID-1125: Add support for FP16 in CLDepthwiseConvolution Change-Id: I4838f5a8e4c33ed646cd05e0bb682fca635a29a3 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/130469 Tested-by: Jenkins Reviewed-by: Michalis Spyrou --- src/core/CL/cl_kernels/depthwise_convolution.cl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/core/CL/cl_kernels/depthwise_convolution.cl') diff --git a/src/core/CL/cl_kernels/depthwise_convolution.cl b/src/core/CL/cl_kernels/depthwise_convolution.cl index 21c28539ef..5f4247e5d3 100644 --- a/src/core/CL/cl_kernels/depthwise_convolution.cl +++ b/src/core/CL/cl_kernels/depthwise_convolution.cl @@ -500,7 +500,7 @@ __kernel void depthwise_weights_reshape( #if defined(HAS_BIAS) if(get_global_id(1) == 0) { - *((__global DATA_TYPE *)(output_ptr + SRC_WIDTH * get_global_size(1) * dst_stride_x)) = *((__global float *)(biases.ptr + get_global_id(2) * biases_stride_x)); + *((__global DATA_TYPE *)(output_ptr + SRC_WIDTH * get_global_size(1) * dst_stride_x)) = *((__global DATA_TYPE *)(biases.ptr + get_global_id(2) * biases_stride_x)); } #endif // defined(HAS_BIAS) } -- cgit v1.2.1