From c80cc3cfe59a6cc8f3113ca84085453d618ae978 Mon Sep 17 00:00:00 2001 From: sunway Date: Thu, 21 Oct 2021 19:12:49 +0800 Subject: [PATCH 1/2] schedule_injective of arm_cpu should consider dtype itemsize --- python/tvm/topi/arm_cpu/injective.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/tvm/topi/arm_cpu/injective.py b/python/tvm/topi/arm_cpu/injective.py index 55f47c5dee4d..330144b33fb6 100644 --- a/python/tvm/topi/arm_cpu/injective.py +++ b/python/tvm/topi/arm_cpu/injective.py @@ -17,6 +17,7 @@ # pylint: disable=invalid-name, unused-variable """Schedule for pooling operators""" import tvm +import numpy as np from tvm import te from ..utils import is_empty_shape @@ -67,7 +68,7 @@ def schedule_injective(outs): if list(s[x].op.axis): # do not vectorize for broadcast - (io, ii) = s[x].split(list(s[x].op.axis)[-1], 4) + (io, ii) = s[x].split(list(s[x].op.axis)[-1], 16 // np.dtype(x.dtype).itemsize) s[x].vectorize(ii) tvm.te.schedule.AutoInlineInjective(s) From 913911294b529ea6c0e7d7372384243ac380c29f Mon Sep 17 00:00:00 2001 From: sunway Date: Mon, 25 Oct 2021 15:40:34 +0800 Subject: [PATCH 2/2] trigger CI