提交 be62dd96 authored 作者: Frederic's avatar Frederic

Don't use toposort when we don't need it.

上级 1c9b5d66
...@@ -1502,7 +1502,7 @@ class GemmOptimizer(Optimizer): ...@@ -1502,7 +1502,7 @@ class GemmOptimizer(Optimizer):
time_toposort = 0 time_toposort = 0
while did_something: while did_something:
t0 = time.time() t0 = time.time()
nodelist = list(fgraph.toposort()) nodelist = theano.gof.graph.io_toposort(fgraph.inputs, fgraph.outputs)
time_toposort += time.time() - t0 time_toposort += time.time() - t0
did_something = False did_something = False
nodelist.reverse() nodelist.reverse()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论