From 729922083c73ff672d4d3c7f8b39ad862963fa6d Mon Sep 17 00:00:00 2001
From: Leyuan Wang <laurawly@gmail.com>
Date: Sat, 18 Nov 2017 21:23:17 -0800
Subject: [PATCH] Fixed nnvm issue #239 (#660)
* scheduler tweaked for super resolution perf
* conv2d_transpose schedule error fixed
* nnvm issue #239 fixed
---
topi/python/topi/cuda/conv2d_nchw.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/topi/python/topi/cuda/conv2d_nchw.py b/topi/python/topi/cuda/conv2d_nchw.py
index a59a99624..0d3f5eedb 100644
--- a/topi/python/topi/cuda/conv2d_nchw.py
+++ b/topi/python/topi/cuda/conv2d_nchw.py
@@ -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:
--
GitLab