提交 61144de5 authored 作者: f0k's avatar f0k

minor fix: GpuConv shape members can be `None`, check for that

上级 ac696286
...@@ -662,7 +662,7 @@ class GpuConv(GpuOp): ...@@ -662,7 +662,7 @@ class GpuConv(GpuOp):
def c_compile_args(self): def c_compile_args(self):
nb = 0 nb = 0
if self.kshp is not None: if (self.kshp is not None) and (self.kshp[1] is not None):
nb = self.kshp[1] nb = self.kshp[1]
return ['-DTHEANO_KERN_WID=' + str(nb)] # ,'-g','-G'] return ['-DTHEANO_KERN_WID=' + str(nb)] # ,'-g','-G']
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论