Unverified Commit 4a39e521 by Animesh Jain Committed by GitHub

[TOPI][x86] Injective schedule improvement (#4786)

* [TOPI][x86] Injective Schedule Improvement.

* Add tiling.

* Vectorize when there is an axis.
parent 60c52e13
...@@ -45,6 +45,12 @@ def schedule_injective_from_existing(sch, out): ...@@ -45,6 +45,12 @@ def schedule_injective_from_existing(sch, out):
sch[out].parallel(fused) sch[out].parallel(fused)
elif len(sch[out].op.axis) >= 1: elif len(sch[out].op.axis) >= 1:
sch[out].parallel(sch[out].op.axis[0]) sch[out].parallel(sch[out].op.axis[0])
# Vectorize the inner most for loop. Tiling first to get a const extent
if len(sch[out].op.axis) >= 1:
l = sch[out].op.axis[-1]
_, li = sch[out].split(l, factor=16)
sch[out].vectorize(li)
return sch return sch
@generic.schedule_injective.register(["cpu"]) @generic.schedule_injective.register(["cpu"])
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment