提交 0df71f39 authored 作者: Roy Xue's avatar Roy Xue

Regression testing

上级 ea5b1c3d
...@@ -40,6 +40,11 @@ def test_profiling(): ...@@ -40,6 +40,11 @@ def test_profiling():
buf = StringIO.StringIO() buf = StringIO.StringIO()
f.profile.summary(buf) f.profile.summary(buf)
# regression testing for future algo speed up
the_string = buf.getvalue()
assert "Max if linker=cvm(default): 8224KB (16408KB)" in the_string
assert "Minimum peak from all valid apply node order is 8208KB" in the_string
finally: finally:
theano.config.profile = old1 theano.config.profile = old1
theano.config.profile_memory = old2 theano.config.profile_memory = old2
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论