From e3af690330c8dc6217109f5228bdda1891d95702 Mon Sep 17 00:00:00 2001
From: dabaiji <yangsijia2@huawei.com>
Date: Mon, 1 Aug 2022 09:38:24 +0800
Subject: [PATCH] auto-tiling opt2: update condition for divisible tile size

---
 src/poly/tiling/tiling_strategy_manager_gpu.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/poly/tiling/tiling_strategy_manager_gpu.cc b/src/poly/tiling/tiling_strategy_manager_gpu.cc
index fbee7d79..6b19a3ec 100644
--- a/src/poly/tiling/tiling_strategy_manager_gpu.cc
+++ b/src/poly/tiling/tiling_strategy_manager_gpu.cc
@@ -1499,7 +1499,7 @@ int64_t GpuStrategy::TileAfterThreadMapping(TileAxis *axis, size_t inner_dim, in
       while (tile_mod % SafeDivisor(tile) != 0 && tile > thread_size) {
         --tile;
       }
-    } else {
+    } else if (axis->forbid_iso) {
       // tile axis with div value
       // e.g. tile cc0 with 512 in the following code (which equals tile floordiv(cc0, 256) with 2)
       // for cc0 in 1024:
-- 
Gitee