diff --git a/mobile/src/io/paddle_mobile.cpp b/mobile/src/io/paddle_mobile.cpp index 74cac25672da1ce1b4534c340616c592de6830c1..95ae3763a2a6415e5fc355ec648633260a5fe411 100644 --- a/mobile/src/io/paddle_mobile.cpp +++ b/mobile/src/io/paddle_mobile.cpp @@ -451,6 +451,8 @@ double PaddleMobile::GetPredictTime() { int input_height = 224; int output_width = 224; int output_height = 224; + int has_group = 0; + int filter_channel = 3; status = clSetKernelArg(kernel, 0, sizeof(int), &c_block); CL_CHECK_ERRORS(status); status = clSetKernelArg(kernel, 1, sizeof(int), &w); @@ -479,6 +481,10 @@ double PaddleMobile::GetPredictTime() { CL_CHECK_ERRORS(status); status = clSetKernelArg(kernel, 13, sizeof(int), &output_height); CL_CHECK_ERRORS(status); + status = clSetKernelArg(kernel, 14, sizeof(int), &filter_channel); + CL_CHECK_ERRORS(status); + status = clSetKernelArg(kernel, 15, sizeof(int), &has_group); + CL_CHECK_ERRORS(status); // cl_event out_event = param.Output()->GetClEvent(); // cl_event wait_event = param.Input()->GetClEvent();