From 8d83482676966aa365ec86d8df97e48e8f7c0160 Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Tue, 5 Dec 2017 18:58:10 +0800
Subject: [PATCH] Fix fusedConv2D bug.

---
 mace/kernels/opencl/fused_conv_2d_opencl.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mace/kernels/opencl/fused_conv_2d_opencl.cc b/mace/kernels/opencl/fused_conv_2d_opencl.cc
index 9e23c57f..86aa0424 100644
--- a/mace/kernels/opencl/fused_conv_2d_opencl.cc
+++ b/mace/kernels/opencl/fused_conv_2d_opencl.cc
@@ -73,9 +73,9 @@ void FusedConv2dFunctor<DeviceType::OPENCL, T>::operator()(const Tensor *input,
   if (kernel_h == kernel_w && kernel_h <= 5 &&
       selector[kernel_h - 1][strides_[0] - 1] != nullptr) {
     auto conv2d_func = selector[kernel_h - 1][strides_[0] - 1];
-    conv2d_func(input, filter, bias, false, paddings.data(), DataTypeToEnum<T>::value, output);
+    conv2d_func(input, filter, bias, true, paddings.data(), DataTypeToEnum<T>::value, output);
   } else {
-    Conv2dOpencl(input, filter, bias, false, strides_[0], paddings.data(), DataTypeToEnum<T>::value, output);
+    Conv2dOpencl(input, filter, bias, true, strides_[0], paddings.data(), DataTypeToEnum<T>::value, output);
   }
 }
 
-- 
GitLab