提交 37edae4a authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add a context hint in local_gpua_careduce in case it's needed.

上级 5df8dbe9
......@@ -621,6 +621,7 @@ def local_gpua_careduce(node, context_name):
node.op.scalar_op, axis=node.op.axis,
dtype=getattr(node.op, 'dtype', None),
acc_dtype=getattr(node.op, 'acc_dtype', None))
x.tag.context_name = context_name
gvar = greduce(x)
# We need to have the make node called, otherwise the mask can
# be None
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论