提交 cd81cf97 authored 作者: Frederic's avatar Frederic

speed up debugmode, less fct call.

上级 b44b0d7a
...@@ -803,15 +803,11 @@ def _check_viewmap(node, storage_map): ...@@ -803,15 +803,11 @@ def _check_viewmap(node, storage_map):
out_alias_idx=other_oi) out_alias_idx=other_oi)
def _is_function_output(node): def _is_used_in_graph(var):
""" """
Returns True if the node in question is the a final output of the graph Returns True if `var` is used by another node in the graph
""" """
return node.clients == [('output', 1)] return not(var.clients == [('output', 1)] or var.clients == [])
def _is_used_in_graph(node):
return not(_is_function_output(node) or node.clients == [])
def _check_strides_match(a, b, warn_err, op): def _check_strides_match(a, b, warn_err, op):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论