This is an automated email from the ASF dual-hosted git repository.
junrushao pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new cddd348 [Fix][TOPI] remove wrong fix in x86's dense_nopack operator
(#8687)
cddd348 is described below
commit cddd3485dd58c1de98c1279a465897a36782befe
Author: Tianqi Zhang (张天启) <[email protected]>
AuthorDate: Mon Aug 16 14:04:08 2021 +0800
[Fix][TOPI] remove wrong fix in x86's dense_nopack operator (#8687)
---
python/tvm/topi/x86/dense.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/python/tvm/topi/x86/dense.py b/python/tvm/topi/x86/dense.py
index 6f2c202..29c378d 100644
--- a/python/tvm/topi/x86/dense.py
+++ b/python/tvm/topi/x86/dense.py
@@ -154,7 +154,6 @@ def _default_dense_nopack_config(cfg, M, N, K):
cfg["tile_k"] = SplitEntity([K // tilek_bn, tilek_bn])
cfg["tile_x"] = SplitEntity([N, 1])
cfg["tile_y"] = SplitEntity([1, M])
- return M, N, K
@autotvm.register_topi_compute("dense_nopack.x86")
@@ -175,7 +174,7 @@ def dense_nopack(cfg, data, weight, bias=None,
out_dtype=None):
"tile_k", 32 if isinstance(K, (tvm.tir.Var, tvm.tir.Any)) else K,
num_outputs=2
)
if cfg.is_fallback:
- M, N, K = _default_dense_nopack_config(cfg, M, N, K)
+ _default_dense_nopack_config(cfg, M, N, K)
vec = cfg["tile_k"].size[-1]
k = te.reduce_axis((0, K // vec), "k")