提交 413bd7ed authored 作者: Reyhane Askari's avatar Reyhane Askari

fix for GpuCAReduceCuda

上级 30335bbd
......@@ -2408,8 +2408,8 @@ def local_gpu_elemwise_careduce(node):
inp = node.inputs[0].owner.inputs[0]
props = node.op._props_dict()
props["pre_scalar_op"] = scalar.basic.sqr
out = GpuCAReduceCuda(**props)(inp)
with inherit_stack_trace(node.outputs):
out = GpuCAReduceCuda(**props)(inp)
return [out]
......
......@@ -40,7 +40,6 @@ def _check_stack_trace(thing):
theano.tensor.elemwise.Elemwise,
theano.ifelse.IfElse,
GpuFromHost, HostFromGpu,
GpuCAReduceCuda,
basic_ops.GpuContiguous,
GpuElemwise,
theano.printing.Print,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论