From 775b4ec3974a8f2c9be15cc0aea6e11addd0983b Mon Sep 17 00:00:00 2001 From: xiebaiyuan Date: Tue, 10 Mar 2020 00:15:34 +0800 Subject: [PATCH] [mobile][opencl] fix conv add bn relu ,test=develop (#3131) --- mobile/src/operators/kernel/cl/conv_add_bn_relu_kernel.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mobile/src/operators/kernel/cl/conv_add_bn_relu_kernel.cpp b/mobile/src/operators/kernel/cl/conv_add_bn_relu_kernel.cpp index 3d1fbc26b8..036ffd5c32 100644 --- a/mobile/src/operators/kernel/cl/conv_add_bn_relu_kernel.cpp +++ b/mobile/src/operators/kernel/cl/conv_add_bn_relu_kernel.cpp @@ -15,7 +15,9 @@ limitations under the License. */ #ifdef FUSION_CONVADDBNRELU_OP #include "operators/kernel/conv_add_bn_relu_kernel.h" + #include + #include "framework/cl/cl_image.h" #include "framework/cl/cl_tool.h" #include "operators/kernel/cl/cl-kernel-func/conv_func.h" @@ -245,7 +247,8 @@ void ConvAddBNReluKernel::Compute( param.NewScale(), param.NewBias()); break; case ConvParam::EXEC_SLIDINGWINDOW3x3_FLOAT: - SWConvAddBnRelu(&this->cl_helper_, param, true, param.Bias()); + SWConvAddBnRelu(&this->cl_helper_, param, true, param.Bias(), + param.NewScale(), param.NewBias()); break; default: PADDLE_MOBILE_THROW_EXCEPTION("Invalid convolution execute mode %d", -- GitLab