提交 055bfd65 authored 作者: Frederic's avatar Frederic

Print more detailled warning.

上级 66c88041
...@@ -670,8 +670,10 @@ def local_gpu_careduce(node): ...@@ -670,8 +670,10 @@ def local_gpu_careduce(node):
if unreshaped_reduce.type == node.outputs[0].type: if unreshaped_reduce.type == node.outputs[0].type:
return [unreshaped_reduce] return [unreshaped_reduce]
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
return False return False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论