提交 812db45f authored 作者: Frederic's avatar Frederic

crash Fix to following change to elemwise on the gpu.

上级 de44d06d
...@@ -815,10 +815,13 @@ nd_collapse_[i]=0; ...@@ -815,10 +815,13 @@ nd_collapse_[i]=0;
""" """
kernels = "".join( kernels = "".join(
[self.c_src_kernel(node, nodename, x) for x in xrange(1, nd + 1)] [self.c_src_kernel(node, nodename, x) for x in xrange(1, nd + 1)]
+ [self.c_src_kernel_Ccontiguous(node, nodename)], + [self.c_src_kernel_Ccontiguous(node, nodename)]
+ [self.c_src_callkernel(node, nodename)]) + [self.c_src_callkernel(node, nodename)])
return defines + kernels return defines + kernels
def c_support_code(self):
raise gof.utils.MethodNotDefined()
def c_code(self, node, nodename, inputs, outputs, sub): def c_code(self, node, nodename, inputs, outputs, sub):
d = dict(sub) d = dict(sub)
nd = node.outputs[0].type.ndim nd = node.outputs[0].type.ndim
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论