conv2d_56_64_128 mark==1 bug fixed (#624)

25847a4f · Leyuan Wang · Tianqi Chen · 08e4d085 · 25847a4f
Commit 25847a4f authored Nov 08, 2017 by Leyuan Wang Committed by Tianqi Chen Nov 08, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

topi/python/topi/cuda/conv2d_nchw.py
+1 -1

No files found.
--- a/topi/python/topi/cuda/conv2d_nchw.py
+++ b/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):
                if mark % i == 0 and vthread_x > 0:
                    num_thread_x = i
                    break
-        if num_thread_x * vthread_x > 128:
+        if mark < 5 or num_thread_x * vthread_x > 128:
            num_thread_x = 8
            vthread_x = 8
        num_thread_y = 8