From d7a748382bdaf430ff906f5595218f050d30fe72 Mon Sep 17 00:00:00 2001
From: Bernhard <62101642+bernhardklein@users.noreply.github.com>
Date: Fri, 13 Mar 2020 22:44:14 +0100
Subject: [PATCH] [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 0945f517970f..79976eb439cb 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),