提交 9410cc30 authored 作者: James Bergstra's avatar James Bergstra

force test_merge_opt_runtime to run in FAST_COMPILE mode because inplace…

force test_merge_opt_runtime to run in FAST_COMPILE mode because inplace optimizations take a really long time on that graph
上级 f93dd747
...@@ -73,7 +73,10 @@ def test_merge_opt_runtime(): ...@@ -73,7 +73,10 @@ def test_merge_opt_runtime():
else: else:
r = x r = x
t = time.time() t = time.time()
f = theano.function([x], r) f = theano.function([x], r, mode='FAST_COMPILE')
# FAST_RUN does in-place optimizer which requires a lot of toposorting, which is actually
# pretty slow at the moment. This test was designed to test MergeOptimizer... so I'm
# leaving toposort optimizations for a later date.
dt = time.time() - t dt = time.time() - t
assert dt < 5.0 #it should never take longer than 5 seconds to compile this graph assert dt < 5.0 #it should never take longer than 5 seconds to compile this graph
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论