From 5b08f7fc38819b1bb551b52db3060c45bd58abed Mon Sep 17 00:00:00 2001 From: xiebaiyuan Date: Sun, 15 Mar 2020 13:17:26 +0800 Subject: [PATCH] [LITE][OPENCL][Image] open depthwise spl ,test=develop (#3186) --- lite/kernels/opencl/conv_image_compute.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lite/kernels/opencl/conv_image_compute.cc b/lite/kernels/opencl/conv_image_compute.cc index 04692d9d75..cda2b82568 100644 --- a/lite/kernels/opencl/conv_image_compute.cc +++ b/lite/kernels/opencl/conv_image_compute.cc @@ -98,7 +98,7 @@ void ConvImageCompute::PrepareForRun() { filter_image_dims[0], filter_image_dims[1], filter_image_v.data()); impl_ = &ConvImageCompute::Conv2d1x1; -// #define DEPTH_CONV_USE_SPL +#define DEPTH_CONV_USE_SPL #ifdef DEPTH_CONV_USE_SPL } else if (filter_dims[1] == 1 && x_dims[1] == output_dims[1] && kernel_h == 3 && kernel_w == 3 && groups > 1) { @@ -126,6 +126,7 @@ void ConvImageCompute::PrepareForRun() { && kernel_h != 3 #endif +#undef DEPTH_CONV_USE_SPL ) { // depth_conv2d kernel_func_names_.push_back("depth_conv2d"); -- GitLab