提交 89a61e75 authored 作者: sentient07's avatar sentient07

Fixed few known mistakes in printing the profile

上级 575fcbee
...@@ -1271,6 +1271,9 @@ class LocalOptGroup(LocalOptimizer): ...@@ -1271,6 +1271,9 @@ class LocalOptGroup(LocalOptimizer):
self.process_count = {} self.process_count = {}
for o in self.opts: for o in self.opts:
self.process_count.setdefault(o, 0)
self.node_created.setdefault(o, 0)
self.time_opts.setdefault(o, 0)
for c in o.tracks(): for c in o.tracks():
self.track_map.setdefault(c, []).append(o) self.track_map.setdefault(c, []).append(o)
...@@ -1299,6 +1302,7 @@ class LocalOptGroup(LocalOptimizer): ...@@ -1299,6 +1302,7 @@ class LocalOptGroup(LocalOptimizer):
def apply_mult_opts(opt_list, node, multiple_opts=False): def apply_mult_opts(opt_list, node, multiple_opts=False):
repl = False repl = False
for opt in opt_list: for opt in opt_list:
opt_start = time.time() opt_start = time.time()
repl = opt.transform(node) repl = opt.transform(node)
...@@ -1307,12 +1311,8 @@ class LocalOptGroup(LocalOptimizer): ...@@ -1307,12 +1311,8 @@ class LocalOptGroup(LocalOptimizer):
continue continue
else: else:
self.time_opts[opt] = opt_start - opt_finish self.time_opts[opt] = opt_start - opt_finish
try:
self.node_created[opt] += len(graph.ops(node.inputs, node.outputs)) self.node_created[opt] += len(graph.ops(node.inputs, node.outputs))
self.process_count[opt] += 1 self.process_count[opt] += 1
except KeyError:
self.node_created.setdefault(opt, 0)
if not multiple_opts or not repl[0].owner: if not multiple_opts or not repl[0].owner:
return repl return repl
assert len(repl) == 1 assert len(repl) == 1
...@@ -1340,6 +1340,7 @@ class LocalOptGroup(LocalOptimizer): ...@@ -1340,6 +1340,7 @@ class LocalOptGroup(LocalOptimizer):
count_opt = [] count_opt = []
not_used = [] not_used = []
not_used_time = 0
for o, count in iteritems(self.process_count): for o, count in iteritems(self.process_count):
if count > 0: if count > 0:
count_opt.append((self.time_opts[o], count, count_opt.append((self.time_opts[o], count,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论