You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by ha...@apache.org on 2020/03/13 21:44:25 UTC
[incubator-tvm] branch master updated: [Relay][AutoTVM] Bug Fix for
ARM CPUs. Lower strict assumption. (#5063)
This is an automated email from the ASF dual-hosted git repository.
haichen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/master by this push:
new d7a7483 [Relay][AutoTVM] Bug Fix for ARM CPUs. Lower strict assumption. (#5063)
d7a7483 is described below
commit d7a748382bdaf430ff906f5595218f050d30fe72
Author: Bernhard <62...@users.noreply.github.com>
AuthorDate: Fri Mar 13 22:44:14 2020 +0100
[Relay][AutoTVM] Bug Fix for ARM CPUs. Lower strict assumption. (#5063)
---
python/tvm/relay/op/strategy/arm_cpu.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python/tvm/relay/op/strategy/arm_cpu.py b/python/tvm/relay/op/strategy/arm_cpu.py
index 0945f51..79976eb 100644
--- a/python/tvm/relay/op/strategy/arm_cpu.py
+++ b/python/tvm/relay/op/strategy/arm_cpu.py
@@ -163,7 +163,7 @@ def conv2d_winograd_without_weight_transfrom_strategy_arm_cpu(attrs, inputs, out
tile_size = attrs.get_int("tile_size")
kh = pad_kh - tile_size + 1
kw = pad_kw - tile_size + 1
- assert kh == 3 and kw == 3 and stride_h == 1 and stride_w == 1
+ assert kh == 3 and kw == 3
strategy.add_implementation(
wrap_compute_conv2d(topi.arm_cpu.conv2d_nchw_winograd),
wrap_topi_schedule(topi.arm_cpu.schedule_conv2d_nchw_winograd),