From 4f4514fee763ec64c6c790ac8871c0e9b2126284 Mon Sep 17 00:00:00 2001 From: liuqi Date: Fri, 26 Jan 2018 15:54:12 +0800 Subject: [PATCH] Change the activation opencl kernel parameter type to float. --- mace/kernels/opencl/cl/activation.cl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mace/kernels/opencl/cl/activation.cl b/mace/kernels/opencl/cl/activation.cl index 02ebff0a..fe8619e2 100644 --- a/mace/kernels/opencl/cl/activation.cl +++ b/mace/kernels/opencl/cl/activation.cl @@ -1,8 +1,8 @@ #include __kernel void activation(__read_only image2d_t input, - __private const DATA_TYPE relux_max_limit, - __private const DATA_TYPE prelu_alpha, + __private const float relux_max_limit, + __private const float prelu_alpha, __write_only image2d_t output) { const int ch_blk = get_global_id(0); const int w = get_global_id(1); -- GitLab