From 7e75d3c55a526dd4d4b1d75070b24645f383e955 Mon Sep 17 00:00:00 2001
From: NazgulLee <lijian30@baidu.com>
Date: Tue, 31 Dec 2019 19:01:35 +0800
Subject: [PATCH] [mobile][opencl] add universal conv_transpose. test=develop
 (#2710)

---
 mobile/src/operators/kernel/cl/conv_transpose_kernel.cpp | 8 +++++++-
 mobile/src/operators/op_param.h                          | 1 +
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/mobile/src/operators/kernel/cl/conv_transpose_kernel.cpp b/mobile/src/operators/kernel/cl/conv_transpose_kernel.cpp
index 8d66b50a99..4261681f3e 100644
--- a/mobile/src/operators/kernel/cl/conv_transpose_kernel.cpp
+++ b/mobile/src/operators/kernel/cl/conv_transpose_kernel.cpp
@@ -43,7 +43,10 @@ bool ConvTransposeKernel<GPU_CL, float>::Init(
     this->cl_helper_.AddKernel("conv_transpose3x3s2",
                                "conv_transpose_kernel.cl");
   } else {
-    PADDLE_MOBILE_THROW_EXCEPTION(" not support ");
+    param->ExecMode() = ConvTransposeParam<GPU_CL>::EXEC_CONVTRANS_FLOAT;
+    param->Filter()->InitConv2dTransposeFilterCLImage(
+        cl_helper_.CLContext(), cl_helper_.CLCommandQueue());
+    this->cl_helper_.AddKernel("conv_transpose", "conv_transpose_kernel.cl");
   }
   return true;
 }
@@ -58,6 +61,9 @@ void ConvTransposeKernel<GPU_CL, float>::Compute(
     case ConvTransposeParam<GPU_CL>::EXEC_CONVTRANS3x3s2_FLOAT:
       ConvTranspose3x3s2AddBnRelu(&this->cl_helper_, param);
       break;
+    case ConvTransposeParam<GPU_CL>::EXEC_CONVTRANS_FLOAT:
+      ConvTransposeAddBnRelu(&this->cl_helper_, param);
+      break;
     default:
       PADDLE_MOBILE_THROW_EXCEPTION(
           "Invalid convolution transpose execute mode %d", param.ExecMode());
diff --git a/mobile/src/operators/op_param.h b/mobile/src/operators/op_param.h
index 85dabe3bcd..f588b9fc79 100644
--- a/mobile/src/operators/op_param.h
+++ b/mobile/src/operators/op_param.h
@@ -2632,6 +2632,7 @@ class ConvTransposeParam : public OpParam {
     EXEC_DECONV4X4_FLOAT,
     EXEC_DEPTHWISETRANS_FLOAT,
     EXEC_CONVTRANS3x3s2_FLOAT,
+    EXEC_CONVTRANS_FLOAT,
   };
 
   ExecMode &ExecMode() const { return exec_mode_; }
-- 
GitLab