提交 c321a11c authored 作者: Roy Xue's avatar Roy Xue

fix some

上级 9a217c5b
...@@ -783,20 +783,20 @@ class ProfileStats(object): ...@@ -783,20 +783,20 @@ class ProfileStats(object):
def min_memory_generator(executables_nodes): def min_memory_generator(executables_nodes):
# print executables_nodes print executables_nodes
for node in list(executables_nodes): for node in list(executables_nodes):
executables_nodes.remove(node) executables_nodes.remove(node)
for var in node.outputs: for var in node.outputs:
compute_map[var][0] = 1 compute_map[var][0] = 1
for var in node.outputs: for var in node.outputs:
for c, _ in var.clients: for c, _ in var.clients:
if c != "output" and check_node_state(c) and c not in executables_nodes: if c != "output" and check_node_state(c):
executables_nodes.append(c) executables_nodes.add(c)
if not executables_nodes: if not executables_nodes:
# executables_nodes.add(node) # executables_nodes.add(node)
yield [node] yield [node]
else: else:
for p in min_memory_generator(executables): for p in min_memory_generator(executables_nodes):
yield [node]+p yield [node]+p
for var in node.outputs: for var in node.outputs:
compute_map[var][0] = 0 compute_map[var][0] = 0
...@@ -835,7 +835,7 @@ class ProfileStats(object): ...@@ -835,7 +835,7 @@ class ProfileStats(object):
i = 0 i = 0
for order in min_memory_generator(node_list): for order in min_memory_generator(executables_nodes):
# print i # print i
i += 1 i += 1
post_thunk_old_storage = [] post_thunk_old_storage = []
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论