Skip to content

Commit

Permalink
[Relay][AutoTVM] Bug Fix for ARM CPUs. Lower strict assumption. (#5063)
Browse files Browse the repository at this point in the history
  • Loading branch information
bernhardklein authored Mar 13, 2020
1 parent 327891c commit d7a7483
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/tvm/relay/op/strategy/arm_cpu.py
Original file line number Diff line number Diff line change
Expand Up @@ -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),
Expand Down

0 comments on commit d7a7483

Please sign in to comment.