You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by ma...@apache.org on 2020/04/21 11:06:22 UTC
[incubator-tvm] branch master updated: [Topi,
ARM] Disbale Winograd for quantized tensors. (#5363)
This is an automated email from the ASF dual-hosted git repository.
masahi 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 b39bd83 [Topi, ARM] Disbale Winograd for quantized tensors. (#5363)
b39bd83 is described below
commit b39bd83172c46c61d64387fd0ee42ea477032200
Author: Animesh Jain <an...@umich.edu>
AuthorDate: Tue Apr 21 04:06:13 2020 -0700
[Topi, ARM] Disbale Winograd for quantized tensors. (#5363)
* [Topi, ARM] Disbale Winograd for quantized tensors.
* Relaxing float
---
python/tvm/relay/op/strategy/arm_cpu.py | 10 ++++++++--
1 file changed, 8 insertions(+), 2 deletions(-)
diff --git a/python/tvm/relay/op/strategy/arm_cpu.py b/python/tvm/relay/op/strategy/arm_cpu.py
index bcef8ab..942d4c7 100644
--- a/python/tvm/relay/op/strategy/arm_cpu.py
+++ b/python/tvm/relay/op/strategy/arm_cpu.py
@@ -59,16 +59,22 @@ def conv2d_strategy_arm_cpu(attrs, inputs, out_type, target):
wrap_compute_conv2d(topi.arm_cpu.conv2d_nchw_spatial_pack),
wrap_topi_schedule(topi.arm_cpu.schedule_conv2d_nchw_spatial_pack),
name="conv2d_nchw_spatial_pack.arm_cpu")
+
# Intel x86 conv2d schedule.
strategy.add_implementation(
wrap_compute_conv2d(topi.x86.conv2d_nchw),
wrap_topi_schedule(topi.x86.schedule_conv2d_nchw),
name="conv2d_nchw.x86")
+
# check if winograd algorithm is applicable
_, _, kh, kw = get_const_tuple(kernel.shape)
pt, pl, pb, pr = topi.nn.get_pad_tuple(padding, (kh, kw))
- if kh == 3 and kw == 3 and stride_h == 1 and stride_w == 1 and \
- dilation_h == 1 and dilation_w == 1:
+ is_winograd_applicable = "float" in data.dtype and \
+ "float" in kernel.dtype and \
+ kh == 3 and kw == 3 and \
+ stride_h == 1 and stride_w == 1 and \
+ dilation_h == 1 and dilation_w == 1
+ if is_winograd_applicable:
strategy.add_implementation(
wrap_compute_conv2d(topi.arm_cpu.conv2d_nchw_winograd),
wrap_topi_schedule(topi.arm_cpu.schedule_conv2d_nchw_winograd),