Skip to content
Snippets Groups Projects
Commit 489ec872 authored by Yuwei HU's avatar Yuwei HU Committed by Tianqi Chen
Browse files

fix fuse reduce_axis error in pooling schedule (#482)

parent acd9db84
No related branches found
No related tags found
No related merge requests found
......@@ -32,9 +32,6 @@ def schedule_global_pool(outs):
Out = outs[0].op.output(0)
s[Pool].set_scope("local")
i, c, h, w = s[Out].op.axis
dh, dw = s[Pool].op.reduce_axis
fuse_index = s[Pool].fuse(dw, dh)
s[Pool].unroll(fuse_index)
by, ty = s[Out].split(i, factor=num_thread)
bx, tx = s[Out].split(c, factor=num_thread)
s[Out].reorder(by, bx, ty, tx)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment