提交 ed049631 authored 作者: sentient07's avatar sentient07

removed an unwanted parameter

上级 c8ee12fe
...@@ -1264,7 +1264,6 @@ class LocalOptGroup(LocalOptimizer): ...@@ -1264,7 +1264,6 @@ class LocalOptGroup(LocalOptimizer):
self.track_map = defaultdict(lambda: []) self.track_map = defaultdict(lambda: [])
assert len(kwargs) == 0 assert len(kwargs) == 0
self.time_opts = {} self.time_opts = {}
self.time_nodes = {}
self.process_count = {} self.process_count = {}
self.applied_true = {} self.applied_true = {}
self.node_created = {} self.node_created = {}
...@@ -1272,7 +1271,6 @@ class LocalOptGroup(LocalOptimizer): ...@@ -1272,7 +1271,6 @@ class LocalOptGroup(LocalOptimizer):
for o in self.opts: for o in self.opts:
self.process_count.setdefault(o, 0) self.process_count.setdefault(o, 0)
self.time_opts.setdefault(o, 0) self.time_opts.setdefault(o, 0)
self.time_nodes.setdefault(o, 0)
self.process_count.setdefault(o, 0) self.process_count.setdefault(o, 0)
self.applied_true.setdefault(o, 0) self.applied_true.setdefault(o, 0)
self.node_created.setdefault(o, 0) self.node_created.setdefault(o, 0)
...@@ -1323,15 +1321,12 @@ class LocalOptGroup(LocalOptimizer): ...@@ -1323,15 +1321,12 @@ class LocalOptGroup(LocalOptimizer):
repl = new_repl repl = new_repl
return repl return repl
node_start = time.time()
new_var = apply_mult_opts(node, node.fgraph, self.apply_all_opts) new_var = apply_mult_opts(node, node.fgraph, self.apply_all_opts)
node_finish = time.time()
self.time_nodes[node] = node_finish - node_start
return new_var return new_var
@staticmethod @staticmethod
def print_profile(stream, prof, level=0): def print_profile(stream, prof, level=0):
(time_opts, time_nodes, process_count, applied_true, node_created) = prof (time_opts, process_count, applied_true, node_created) = prof
blanc = (' ' * int(level)) blanc = (' ' * int(level))
print(blanc, "LocalOptGroup", file=stream) print(blanc, "LocalOptGroup", file=stream)
print(blanc, "---------------------", file=stream) print(blanc, "---------------------", file=stream)
...@@ -2059,7 +2054,6 @@ class TopoOptimizer(NavigatorOptimizer): ...@@ -2059,7 +2054,6 @@ class TopoOptimizer(NavigatorOptimizer):
print(blanc, " callback_time", callback_time, file=stream) print(blanc, " callback_time", callback_time, file=stream)
if isinstance(lopt, LocalOptGroup): if isinstance(lopt, LocalOptGroup):
lopt.print_profile(stream, (lopt.time_opts, lopt.print_profile(stream, (lopt.time_opts,
lopt.time_nodes,
lopt.process_count, lopt.process_count,
lopt.applied_true, lopt.applied_true,
lopt.node_created), lopt.node_created),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论