diff --git a/mace/kernels/opencl/matmul.cc b/mace/kernels/opencl/matmul.cc index 7e5e52b43614eb51eb7b181c92ef4aea0f8ff612..4df9d58d74098bd6aae2e9697333090f05e215ba 100644 --- a/mace/kernels/opencl/matmul.cc +++ b/mace/kernels/opencl/matmul.cc @@ -82,7 +82,7 @@ MaceStatus MatMulFunctor::operator()(const Tensor *A, kernel_.setArg(idx++, static_cast(height_blocks)); kernel_.setArg(idx++, static_cast(RoundUpDiv4(K))); - const std::vector lws = {kwg_size_ / 64, 64, 1}; + const std::vector lws = {kwg_size_ / 64, 64, 0}; std::string tuning_key = Concat("matmul_opencl_kernel", batch, height, width); MACE_RETURN_IF_ERROR(TuningOrRun2DKernel(kernel_, tuning_key, gws, lws, future));