Skip to content
Snippets Groups Projects
Commit 72992208 authored by Leyuan Wang's avatar Leyuan Wang Committed by Tianqi Chen
Browse files

Fixed nnvm issue #239 (#660)

* scheduler tweaked for super resolution perf

* conv2d_transpose schedule error fixed

* nnvm issue #239 fixed
parent 3479b9ab
No related branches found
No related tags found
No related merge requests found
......@@ -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:
......
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