fix conv1x1 tune bug test=develop (#6879)

PaddlePaddle · Sep 8, 2021 · 1e4a3fb · 1e4a3fb
1 parent 557b9ee
commit 1e4a3fb
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/lite/kernels/opencl/conv_image_compute.cc b/lite/kernels/opencl/conv_image_compute.cc
@@ -821,6 +821,13 @@ void ConvImageCompute::SetLocalWorkSize(size_t repeats /*=4*/) {
                                       static_cast<size_t>(tuned_in_map[5])};
       int func_id = tuned_in_map[6];
       if (func_id == 0) {
+        context.cl_context()->AddKernel(kernel_func_names_[0],
+                                        kernel_func_paths_[0],
+                                        build_options_[0],
+                                        time_stamp_);
+        kernel_key.str("");
+        kernel_key << kernel_func_names_[0] << build_options_[0] << time_stamp_;
+        kernel_ = context.cl_context()->GetKernel(kernel_key.str());
         return;
       } else if (func_id == 1) {
         kernel_func_names_[0] = "conv2d_1x1_h1w5c1";