Commit 6f355762 by Yizhi Liu Committed by Tianqi Chen

fix x86 pooling topi schedule (#1132)

parent 8e9eaae5
......@@ -9,8 +9,7 @@ def _parallel_sch(sch):
fused = sch.fuse(sch.op.axis[0], sch.op.axis[1], sch.op.axis[2])
sch.parallel(fused)
elif len(sch.op.axis) >= 3:
n, c, _, _ = sch.op.axis
fused = sch.fuse(n, c) # for nhwc layout, fuse n and h
fused = sch.fuse(sch.op.axis[0], sch.op.axis[1])
sch.parallel(fused)
else:
sch.parallel(sch.op.axis[0])
......
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