提交 637dd104 authored 作者: Roy Xue's avatar Roy Xue

Using DONE_SET method to make algorithm faster

上级 ee36aaae
...@@ -797,6 +797,7 @@ class ProfileStats(object): ...@@ -797,6 +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([])
# 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
compute_map = defaultdict(lambda: [0]) compute_map = defaultdict(lambda: [0])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论