Skip to content

Commit ca036d3

Browse files
KnowingNothingzhengsize
authored andcommitted
[FIX][TOPI][X86] schedule dense pack
1 parent 9384353 commit ca036d3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

topi/python/topi/x86/dense.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,8 +59,8 @@ def _declaration_dense_pack(cfg, data, weight, bias=None, out_dtype=None):
5959
M, K = get_const_tuple(data.shape) # batch, in_dim
6060
N, _ = get_const_tuple(weight.shape) # out_dim
6161
# create tuning space
62-
cfg.define_split("tile_y", 32 if isinstance(M, tvm.expr.Var) else M, num_outputs=2)
63-
cfg.define_split("tile_x", 32 if isinstance(N, tvm.expr.Var) else N, num_outputs=2)
62+
cfg.define_split("tile_y", 32 if isinstance(M, tvm.expr.Var) else M, num_outputs=3)
63+
cfg.define_split("tile_x", 32 if isinstance(N, tvm.expr.Var) else N, num_outputs=3)
6464
cfg.define_split("tile_k", 32 if isinstance(K, tvm.expr.Var) else K, num_outputs=2)
6565
if cfg.is_fallback:
6666
_default_dense_pack_config(cfg, M, N, K)

0 commit comments

Comments
 (0)