提交 a0171bcd authored 作者: Frederic's avatar Frederic

fix compilation crash recently introduced on the cuda code.

上级 c4e182ab
...@@ -3362,7 +3362,7 @@ class Rebroadcast(Op): ...@@ -3362,7 +3362,7 @@ class Rebroadcast(Op):
#TODO: if your type is not listed here, make a damn registry of #TODO: if your type is not listed here, make a damn registry of
# shape_i ops for various types of variables. # shape_i ops for various types of variables.
# Do not continue this madness. # Do not continue this madness.
return super(Rebroadcast, self).c_code(node, nodename, (x,), (out,), sub) return super(Rebroadcast, self).c_code(node, nodename, inp, out, sub)
def c_code_cache_version(self): def c_code_cache_version(self):
return (1,) return (1,)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论