提交 64fb4fe6 编写于 作者: R Ray Liu 提交者: GitHub

Merge pull request #1033 from codeWorm2015/opencl

update climage
......@@ -45,6 +45,10 @@ class CLImage {
return {};
}
cl_mem GetCLImage() {
return cl_image_;
}
private:
bool initialized_ = false;
cl_mem cl_image_;
......
......@@ -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);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册