From 641024f5e39d97291c6614c2624088b70f10ce6a Mon Sep 17 00:00:00 2001 From: deepIgnorance Date: Fri, 27 Dec 2019 02:10:44 +0800 Subject: [PATCH] [FIX][TOPI][X86] schedule dense pack (#4539) --- topi/python/topi/x86/dense.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/topi/python/topi/x86/dense.py b/topi/python/topi/x86/dense.py index 605a175..b7a3d6d 100644 --- a/topi/python/topi/x86/dense.py +++ b/topi/python/topi/x86/dense.py @@ -59,8 +59,8 @@ def _declaration_dense_pack(cfg, data, weight, bias=None, out_dtype=None): M, K = get_const_tuple(data.shape) # batch, in_dim N, _ = get_const_tuple(weight.shape) # out_dim # create tuning space - cfg.define_split("tile_y", 32 if isinstance(M, tvm.expr.Var) else M, num_outputs=2) - cfg.define_split("tile_x", 32 if isinstance(N, tvm.expr.Var) else N, num_outputs=2) + cfg.define_split("tile_y", 32 if isinstance(M, tvm.expr.Var) else M, num_outputs=3) + cfg.define_split("tile_x", 32 if isinstance(N, tvm.expr.Var) else N, num_outputs=3) cfg.define_split("tile_k", 32 if isinstance(K, tvm.expr.Var) else K, num_outputs=2) if cfg.is_fallback: _default_dense_pack_config(cfg, M, N, K) -- 2.7.4