提交 db00bc92 编写于 作者: Y yangfei

imp add op kernel

上级 97fc0e68
...@@ -64,7 +64,7 @@ void ElementwiseAddKernel<GPU_CL, float>::Compute( ...@@ -64,7 +64,7 @@ void ElementwiseAddKernel<GPU_CL, float>::Compute(
cl_mem input_image = input->GetCLImage(); cl_mem input_image = input->GetCLImage();
cl_mem bias_image = bias->GetCLImage(); cl_mem bias_image = bias->GetCLImage();
cl_mem output_image = output->GetCLImage(); cl_mem output_image = output->GetCLImage();
int tensor_w = input->dims()[4]; int tensor_w = input->dims()[3];
status = clSetKernelArg(kernel, 0, sizeof(cl_mem), (void *)&input_image); status = clSetKernelArg(kernel, 0, sizeof(cl_mem), (void *)&input_image);
CL_CHECK_ERRORS(status); CL_CHECK_ERRORS(status);
status = clSetKernelArg(kernel, 1, sizeof(cl_mem), (void *)&bias_image); status = clSetKernelArg(kernel, 1, sizeof(cl_mem), (void *)&bias_image);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册