提交 03c8d8fd authored 作者: Roy Xue's avatar Roy Xue

Remove check_node_state

上级 7c7ce9a2
......@@ -779,23 +779,13 @@ class ProfileStats(object):
if isinstance(val, graph.Constant):
compute_map[val][0] = 1
def check_node_state(node):
"""
Check if an Apply node is valid(has inputs).
:param node: Apply Node
"""
inputs = node.inputs
outputs = node.outputs
deps = inputs + node.destroy_dependencies
computed_ins = all(compute_map[v][0] for v in deps)
return computed_ins
# Initial executable_nodes
executable_nodes = set()
for var in fgraph.inputs:
for c, _ in var.clients:
if c != "output" and check_node_state(c):
if c != "output":
deps = c.inputs + c.destroy_dependencies
if all(compute_map[v][0] for v in deps):
executable_nodes.add(c)
def min_memory_generator(executable_nodes, viewed_by, view_of):
......@@ -893,7 +883,9 @@ class ProfileStats(object):
for var in node.outputs:
for c, _ in var.clients:
if c != "output" and check_node_state(c):
if c != "output":
deps = c.inputs + c.destroy_dependencies
if all(compute_map[v][0] for v in deps):
new_exec_nodes.add(c)
if not new_exec_nodes:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论