Commit 72992208 by Leyuan Wang Committed by Tianqi Chen

Fixed nnvm issue #239 (#660)

* scheduler tweaked for super resolution perf

* conv2d_transpose schedule error fixed

* nnvm issue #239 fixed
parent 3479b9ab
......@@ -75,6 +75,9 @@ def conv2d_56_64_128(s, temp, temp_R, temp_S, Filter_S, Out, Out_L, flag):
if mark % 8 == 0 and mark % 7 == 0:
num_thread_x = 8
vthread_x = 7
elif mark % 4 == 0 and mark % 7 == 0:
num_thread_x = 4
vthread_x = 7
else:
for i in range(5, mark):
if mark % i == 0 and num_thread_x == 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