Skip to content

Commit

Permalink
[OpenCL] fix mali gpu n>1 conv bug (#9958)
Browse files Browse the repository at this point in the history
  • Loading branch information
sprouteer authored Feb 10, 2023
1 parent ede9f46 commit 8c493f5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion lite/kernels/opencl/conv_image_compute.cc
Original file line number Diff line number Diff line change
Expand Up @@ -238,7 +238,7 @@ void ConvImageCompute::PrepareForRun() {
}
#endif
} else if (is_mali_ && filter_tensor_h_ == 1 && filter_tensor_w_ == 1 &&
groups_ == 1) { // mali conv1x1
groups_ == 1 && input_tensor_n_ == 1) { // mali conv1x1
if (task_size <= threshold_2) {
CLImageConverterNBlock converter;
kernel_func_names_.push_back("conv2d_1x1_mali_h1w2c1");
Expand Down

0 comments on commit 8c493f5

Please sign in to comment.