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

Change the DONE_SET to done_set

上级 b1cae5e5
...@@ -797,7 +797,7 @@ class ProfileStats(object): ...@@ -797,7 +797,7 @@ class ProfileStats(object):
mem_bound = numpy.inf mem_bound = numpy.inf
# This take only the inputs/outputs dependencies. # This take only the inputs/outputs dependencies.
dependencies = fgraph.profile.dependencies dependencies = fgraph.profile.dependencies
DONE_SET = set([]) done_set = set([])
done_dict = {} done_dict = {}
# Initial compute_map which is used to check if a node is valid # Initial compute_map which is used to check if a node is valid
...@@ -911,8 +911,8 @@ class ProfileStats(object): ...@@ -911,8 +911,8 @@ class ProfileStats(object):
mem_count -= mem_freed mem_count -= mem_freed
DONE_SET.add(node) done_set.add(node)
frozen_set = frozenset(DONE_SET) frozen_set = frozenset(done_set)
if frozen_set not in done_dict: if frozen_set not in done_dict:
done_dict[frozen_set] = mem_count done_dict[frozen_set] = mem_count
else: else:
...@@ -920,7 +920,7 @@ class ProfileStats(object): ...@@ -920,7 +920,7 @@ class ProfileStats(object):
if past_mem > mem_count: if past_mem > mem_count:
done_dict[frozen_set] = mem_count done_dict[frozen_set] = mem_count
else: else:
DONE_SET.remove(node) done_set.remove(node)
mem_count -= mem_created mem_count -= mem_created
max_mem_count = max_storage max_mem_count = max_storage
mem_count += mem_freed mem_count += mem_freed
...@@ -955,7 +955,7 @@ class ProfileStats(object): ...@@ -955,7 +955,7 @@ class ProfileStats(object):
min_memory_generator(new_exec_nodes, viewed_by, view_of) min_memory_generator(new_exec_nodes, viewed_by, view_of)
# Reset track variables # Reset track variables
DONE_SET.remove(node) done_set.remove(node)
mem_count -= mem_created mem_count -= mem_created
max_mem_count = max_storage max_mem_count = max_storage
mem_count += mem_freed mem_count += mem_freed
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论