From a36ccf1b5c66a5f25bdde1ce69b60e49ed67bb41 Mon Sep 17 00:00:00 2001 From: Moritz Pflanzer Date: Mon, 24 Jul 2017 15:05:12 +0100 Subject: COMPMID-417: Fix CL F16 ActivationLayer Change-Id: I5b703d59661f6b50e3de4b2313be381b8d27030a Reviewed-on: http://mpd-gerrit.cambridge.arm.com/81951 Tested-by: Kaizen Reviewed-by: Anthony Barbier --- src/core/CL/cl_kernels/activation_layer.cl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/core/CL') diff --git a/src/core/CL/cl_kernels/activation_layer.cl b/src/core/CL/cl_kernels/activation_layer.cl index 9f958610d6..a1e7002c8b 100644 --- a/src/core/CL/cl_kernels/activation_layer.cl +++ b/src/core/CL/cl_kernels/activation_layer.cl @@ -59,7 +59,7 @@ // Logistic Activation inline TYPE logistic_op(TYPE x) { - return DIV_OP(CONST_ONE, ADD_OP(CONST_ONE, EXP_OP(-x))); + return DIV_OP((TYPE)CONST_ONE, ADD_OP((TYPE)CONST_ONE, EXP_OP(-x))); } // Hyperbolic Tangent Activation inline TYPE tanh_op(TYPE x) @@ -84,7 +84,7 @@ inline TYPE lrelu_op(TYPE x) // Soft RELU Activation inline TYPE srelu_op(TYPE x) { - return LOG_OP(ADD_OP(CONST_ONE, EXP_OP(x))); + return LOG_OP(ADD_OP((TYPE)CONST_ONE, EXP_OP(x))); } // Absolute Activation inline TYPE abs_op(TYPE x) -- cgit v1.2.1