提交 0282eac7 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix opt crash/warning. The dtype props wasn't passed

上级 e679cd14
...@@ -1903,10 +1903,10 @@ def local_gpu_elemwise_careduce(node): ...@@ -1903,10 +1903,10 @@ def local_gpu_elemwise_careduce(node):
isinstance(node.inputs[0].owner.op.scalar_op, scalar.basic.Sqr)): isinstance(node.inputs[0].owner.op.scalar_op, scalar.basic.Sqr)):
op = node.op op = node.op
inp = node.inputs[0].owner.inputs[0] inp = node.inputs[0].owner.inputs[0]
return [GpuCAReduceCuda(scalar_op=op.scalar_op, props = node.op._props_dict()
axis=op.axis, props["pre_scalar_op"] = scalar.basic.sqr
reduce_mask=op.reduce_mask, out = GpuCAReduceCuda(**props)(inp)
pre_scalar_op=scalar.basic.sqr)(inp)] return [out]
@local_optimizer(None) @local_optimizer(None)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论