提交 c47c2209 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix conv op compilation when the c_code fct was changing the signature of the c…

fix conv op compilation when the c_code fct was changing the signature of the c code. That is not allowed.
上级 aa2b5a43
...@@ -423,6 +423,9 @@ class ConvOp(Op): ...@@ -423,6 +423,9 @@ class ConvOp(Op):
self.unroll_kern=unroll_kern self.unroll_kern=unroll_kern
self.unroll_patch=unroll_patch self.unroll_patch=unroll_patch
if self.unroll_batch and not self.unroll_kern: self.unroll_kern = 1
if self.unroll_kern and not self.unroll_batch: self.unroll_batch = 1
#downcast unroll_batch if not a divisor of batch size #downcast unroll_batch if not a divisor of batch size
if self.unroll_batch>0 and self.bsize % self.unroll_batch!=0: if self.unroll_batch>0 and self.bsize % self.unroll_batch!=0:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论