提交 143dd814 authored 作者: Frederic's avatar Frederic

Remove printing during test

上级 64eb0117
...@@ -246,7 +246,7 @@ AddConfigVar( ...@@ -246,7 +246,7 @@ AddConfigVar(
'on_opt_error', 'on_opt_error',
("What to do when an optimization crashes: warn and skip it, raise " ("What to do when an optimization crashes: warn and skip it, raise "
"the exception, or fall into the pdb debugger."), "the exception, or fall into the pdb debugger."),
EnumStr('warn', 'raise', 'pdb'), EnumStr('warn', 'raise', 'pdb', 'ignore'),
in_c_key=False) in_c_key=False)
......
...@@ -1364,9 +1364,10 @@ class NavigatorOptimizer(Optimizer): ...@@ -1364,9 +1364,10 @@ class NavigatorOptimizer(Optimizer):
def warn(exc, nav, repl_pairs, local_opt): def warn(exc, nav, repl_pairs, local_opt):
"""failure_callback for NavigatorOptimizer: print traceback """failure_callback for NavigatorOptimizer: print traceback
""" """
_logger.error("Optimization failure due to: %s" % str(local_opt)) if config.on_opt_error != 'ignore':
_logger.error("TRACEBACK:") _logger.error("Optimization failure due to: %s" % str(local_opt))
_logger.error(traceback.format_exc()) _logger.error("TRACEBACK:")
_logger.error(traceback.format_exc())
if config.on_opt_error == 'pdb': if config.on_opt_error == 'pdb':
pdb.post_mortem(sys.exc_info()[2]) pdb.post_mortem(sys.exc_info()[2])
elif isinstance(exc, AssertionError) or config.on_opt_error == 'raise': elif isinstance(exc, AssertionError) or config.on_opt_error == 'raise':
......
...@@ -106,15 +106,17 @@ def test_local_assert_no_cpu_op(): ...@@ -106,15 +106,17 @@ def test_local_assert_no_cpu_op():
mode_local_assert = mode_local_assert.excluding("local_gpu_elemwise_1") mode_local_assert = mode_local_assert.excluding("local_gpu_elemwise_1")
old = config.assert_no_cpu_op old = config.assert_no_cpu_op
old2 = config.on_opt_error
# If the flag is raise # If the flag is raise
try: try:
config.assert_no_cpu_op = 'raise' config.assert_no_cpu_op = 'raise'
config.on_opt_error = 'ignore'
assert_raises(AssertionError, theano.function, assert_raises(AssertionError, theano.function,
[], out, mode=mode_local_assert) [], out, mode=mode_local_assert)
finally: finally:
config.assert_no_cpu_op = old config.assert_no_cpu_op = old
config.on_opt_error = old2
# If the flag is ignore # If the flag is ignore
try: try:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论