Skip to content

T
tic
Commit 25847a4f by Leyuan Wang Committed by Tianqi Chen
Options

conv2d_56_64_128 mark==1 bug fixed (#624)

parent 08e4d085
Showing with 1 additions and 1 deletions
topi/python/topi/cuda/conv2d_nchw.py
...@@ -80,7 +80,7 @@ def conv2d_56_64_128(s, temp, temp_R, temp_S, Filter_S, Out, Out_L, flag): ...@@ -80,7 +80,7 @@ def conv2d_56_64_128(s, temp, temp_R, temp_S, Filter_S, Out, Out_L, flag):
if mark % i == 0 and vthread_x > 0: if mark % i == 0 and vthread_x > 0:
num_thread_x = i num_thread_x = i
break break
if num_thread_x * vthread_x > 128: if mark < 5 or num_thread_x * vthread_x > 128:
num_thread_x = 8 num_thread_x = 8
vthread_x = 8 vthread_x = 8
num_thread_y = 8 num_thread_y = 8
......
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