diff --git a/python/tvm/topi/adreno/pooling.py b/python/tvm/topi/adreno/pooling.py index 24d8675d1a49..49f103c04a2f 100644 --- a/python/tvm/topi/adreno/pooling.py +++ b/python/tvm/topi/adreno/pooling.py @@ -62,6 +62,7 @@ def _schedule(PaddedInput, Pool): s[OL].vectorize(s[OL].op.axis[-1]) else: s[Pool].compute_at(s[Out], tx) + s[Pool].vectorize(s[Pool].op.axis[-1]) scheduled_ops = []