提交 5956b40d authored 作者: Frederic's avatar Frederic

Fix test in debug mode

上级 143dd814
...@@ -4261,7 +4261,13 @@ class T_Scan(unittest.TestCase): ...@@ -4261,7 +4261,13 @@ class T_Scan(unittest.TestCase):
# There should be 3 outputs greater than 10: prior_result[0] at step 3, # There should be 3 outputs greater than 10: prior_result[0] at step 3,
# and prior_result[1] at steps 2 and 3. # and prior_result[1] at steps 2 and 3.
assert detect_large_outputs.large_count == 3 if theano.config.mode in ["DEBUG_MODE", "DebugMode"]:
# DebugMode will run all the intermediate nodes, so we
# should expect a multiple of 3, not exactly 3.
assert detect_large_outputs.large_count % 3 == 0
else:
assert detect_large_outputs.large_count == 3
class ScanGpuTests: class ScanGpuTests:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论