提交 09bd74b1 authored 作者: Frederic's avatar Frederic

Better warning message.

上级 8a52394d
...@@ -630,8 +630,10 @@ def local_gpu_careduce(node): ...@@ -630,8 +630,10 @@ def local_gpu_careduce(node):
if rval.type == node.outputs[0].type: if rval.type == node.outputs[0].type:
return [rval] return [rval]
else: else:
print >> sys.stderr, \ print >> sys.stderr, (
"WARNING: local_gpu_careduce got type wrong" "WARNING: local_gpu_careduce got type wrong",
unreshaped_reduce.type, node.outputs[0].type,
node)
return None return None
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论