提交 5d23dc6f authored 作者: Reyhane Askari's avatar Reyhane Askari

minor fix

上级 08762910
......@@ -2407,7 +2407,7 @@ def local_gpu_elemwise_careduce(node):
props["pre_scalar_op"] = scalar.basic.sqr
out = GpuCAReduceCuda(**props)(inp)
with inherit_stack_trace(node.outputs):
return out
return [out]
@local_optimizer(None)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论