You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by lm...@apache.org on 2021/01/22 18:38:35 UTC
[tvm] 01/01: [AutoScheduler] Fix conv3d's op strategy for
auto-scheduler
This is an automated email from the ASF dual-hosted git repository.
lmzheng pushed a commit to branch merrymercy-patch-1
in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 676026a959427d1f05ef7ab44886d5cb98ad2d10
Author: Lianmin Zheng <li...@gmail.com>
AuthorDate: Fri Jan 22 10:38:13 2021 -0800
[AutoScheduler] Fix conv3d's op strategy for auto-scheduler
---
python/tvm/relay/op/strategy/x86.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python/tvm/relay/op/strategy/x86.py b/python/tvm/relay/op/strategy/x86.py
index 9e3e191..edfaaee 100644
--- a/python/tvm/relay/op/strategy/x86.py
+++ b/python/tvm/relay/op/strategy/x86.py
@@ -304,7 +304,7 @@ def conv3d_strategy_cpu(attrs, inputs, out_type, target):
# or packed layouts.
if layout == "NCDHW":
strategy.add_implementation(
- wrap_compute_conv3d(topi.nn.conv3d_ncdhw, need_auto_scheduler_layout=True),
+ wrap_compute_conv3d(topi.nn.conv3d_ncdhw),
naive_schedule,
name="conv3d_ncdhw.x86",
)