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 <[email protected]> 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", )
