Add vectorization into some adreno pool flow

Co-authored-by: Li <[email protected]>
apache · May 6, 2022 · 21e85d0 · 21e85d0
1 parent 31368e4
commit 21e85d0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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 = []