diff --git a/mindspore/lite/src/runtime/kernel/opencl/opencl_kernel.cc b/mindspore/lite/src/runtime/kernel/opencl/opencl_kernel.cc index 5df0f217650..a9801021f5a 100644 --- a/mindspore/lite/src/runtime/kernel/opencl/opencl_kernel.cc +++ b/mindspore/lite/src/runtime/kernel/opencl/opencl_kernel.cc @@ -338,7 +338,9 @@ std::set OpenCLKernel::GenerateLocalByGlobal(size_t global_i) { std::set local_ = {}; int index = 1; while (index <= global_i) { - local_.insert(index); + if (global_i % index == 0) { + local_.insert(index); + } index *= 2; } for (size_t i = 1; i <= 16; i++) {