提交 52baaf7d authored 作者: sentient07's avatar sentient07

A small optimization

上级 e8017096
......@@ -265,6 +265,11 @@ class GraphToGPU(NavigatorOptimizer):
time_opts = {}
node_created = {}
process_count = {}
t_topo = time.time()
topo = fgraph.toposort()
time_topo = time.time()
toposort_timing = time_topo - t_topo
# Building a new graph
# Iterating through inputs of graph
for i in fgraph.inputs:
......@@ -275,7 +280,7 @@ class GraphToGPU(NavigatorOptimizer):
for i in fgraph.variables:
if isinstance(i, theano.Constant):
mapping[i] = i
for node in fgraph.toposort():
for node in topo:
for lopt in (self.local_optimizers_map.get(node.op, []) +
self.local_optimizers_map.get(type(node.op), []) +
self.local_optimizers_all):
......@@ -283,11 +288,6 @@ class GraphToGPU(NavigatorOptimizer):
time_opts.setdefault(lopt, 0)
node_created.setdefault(lopt, 0)
t_topo = time.time()
topo = fgraph.toposort()
time_topo = time.time() - t_topo
toposort_timing = time_topo - t_topo
for node in topo:
if isinstance(node.op, HostFromGpu):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论