提交 29416c10 authored 作者: Frederic Bastien's avatar Frederic Bastien 提交者: Reyhane Askari

Update test.

上级 3d558c04
......@@ -56,8 +56,8 @@ class Test_profiling(unittest.TestCase):
lines1 = [l for l in the_string.split("\n") if "Max if linker" in l]
lines2 = [l for l in the_string.split("\n") if "Minimum peak" in l]
if theano.config.device == 'cpu':
assert "CPU: 4112KB (8204KB)" in the_string, (lines1, lines2)
assert "CPU: 8204KB (12296KB)" in the_string, (lines1, lines2)
assert "CPU: 4112KB (4112KB)" in the_string, (lines1, lines2)
assert "CPU: 8204KB (8204KB)" in the_string, (lines1, lines2)
assert "CPU: 8208KB" in the_string, (lines1, lines2)
assert "Minimum peak from all valid apply node order is 4104KB" in the_string, (
lines1, lines2)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论