From f3d234809fa853237e118885784d5ca2413935f9 Mon Sep 17 00:00:00 2001
From: liuruilong <liuruilong@baidu.com>
Date: Sat, 29 Sep 2018 16:11:26 +0800
Subject: [PATCH] update climage

---
 src/framework/cl/cl_image.h             | 4 ++++
 src/operators/kernel/cl/conv_kernel.cpp | 1 -
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/framework/cl/cl_image.h b/src/framework/cl/cl_image.h
index d802e4e6dc..647a93d7fc 100644
--- a/src/framework/cl/cl_image.h
+++ b/src/framework/cl/cl_image.h
@@ -45,6 +45,10 @@ class CLImage {
     return {};
   }
 
+  cl_mem GetCLImage() {
+    return cl_image_;
+  }
+
  private:
   bool initialized_ = false;
   cl_mem cl_image_;
diff --git a/src/operators/kernel/cl/conv_kernel.cpp b/src/operators/kernel/cl/conv_kernel.cpp
index 419eb51f50..451f96447e 100644
--- a/src/operators/kernel/cl/conv_kernel.cpp
+++ b/src/operators/kernel/cl/conv_kernel.cpp
@@ -28,7 +28,6 @@ bool ConvKernel<GPU_CL, float>::Init(ConvParam<GPU_CL> *param) {
 
 template <>
 void ConvKernel<GPU_CL, float>::Compute(const ConvParam<GPU_CL> &param) {
-
   auto kernel = this->cl_helper_.KernelAt(0);
   size_t global_work_size[3] = {1, 2, 3};
   clEnqueueNDRangeKernel(this->cl_helper_.CLCommandQueue(), kernel, 3, NULL, global_work_size, NULL, 0, NULL, NULL);
-- 
GitLab