diff --git a/src/framework/cl/cl_image.h b/src/framework/cl/cl_image.h index d802e4e6dc24c5beee959a808a34da05c5ced0a7..647a93d7fcc4d56809e09bb5258a18407a6598ba 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 419eb51f5053bd93177dc8c1f9613f5d512ce79a..451f96447e458524e698f5068bba73f763b236b2 100644 --- a/src/operators/kernel/cl/conv_kernel.cpp +++ b/src/operators/kernel/cl/conv_kernel.cpp @@ -28,7 +28,6 @@ bool ConvKernel::Init(ConvParam *param) { template <> void ConvKernel::Compute(const ConvParam ¶m) { - 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);