From 4512aecb7945825fa84dbca488a0c84cd780a94b Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Wed, 18 Jul 2018 13:10:38 +0800
Subject: [PATCH] Fix the bug: tuning 2d lws for gpu do not work.

---
 mace/kernels/opencl/helper.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mace/kernels/opencl/helper.cc b/mace/kernels/opencl/helper.cc
index 78e50871..009bf276 100644
--- a/mace/kernels/opencl/helper.cc
+++ b/mace/kernels/opencl/helper.cc
@@ -408,7 +408,7 @@ void TuningOrRun2DKernel(const cl::Kernel &kernel,
         {kwg_size / 128, 128, 0}, {kwg_size / 256, 256, 0},
         {kwg_size, 1, 0},         {1, kwg_size, 0}};
     for (auto &ele : candidates) {
-      const uint32_t tmp = ele[0] * ele[1] * ele[2];
+      const uint32_t tmp = ele[0] * ele[1];
       if (0 < tmp && tmp <= kwg_size) {
         results.push_back(ele);
       }
-- 
GitLab