提交 58b3b4cf 编写于 作者: J Jiaying Zhao 提交者: GitHub

Merge pull request #1204 from smilejames/opencl

revert opencl conv3x3 kernel
......@@ -150,7 +150,7 @@ bool ConvAddBNReluKernel<GPU_CL, float>::Init(
} else if (param->Filter()->dims()[2] == 3 &&
param->Filter()->dims()[3] == 3) {
param->Filter()->InitNImage(cl_helper_.CLContext(),
param->Filter()->InitCLImage(cl_helper_.CLContext(),
cl_helper_.CLCommandQueue());
this->cl_helper_.AddKernel("conv_3x3", "conv_add_bn_relu_kernel.cl");
......
......@@ -45,7 +45,7 @@ bool ConvAddKernel<GPU_CL, float>::Init(FusionConvAddParam<GPU_CL> *param) {
} else if (param->Filter()->dims()[2] == 3 &&
param->Filter()->dims()[3] == 3) {
param->Filter()->InitNImage(cl_helper_.CLContext(),
param->Filter()->InitCLImage(cl_helper_.CLContext(),
cl_helper_.CLCommandQueue());
this->cl_helper_.AddKernel("conv_3x3", "conv_add_kernel.cl");
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册