提交 26d43bf0 authored 作者: Caglar's avatar Caglar

fixed the assertion raised.

上级 a7b22194
...@@ -100,13 +100,12 @@ def test_local_assert_no_cpu_op(): ...@@ -100,13 +100,12 @@ def test_local_assert_no_cpu_op():
mode_local_assert = mode_with_gpu.including("local_assert_no_cpu_op") mode_local_assert = mode_with_gpu.including("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
# If the flag is raise # If the flag is raise
try: try:
config.assert_no_cpu_op = 'raise' config.assert_no_cpu_op = 'raise'
assert_raises(AssertionError, theano.function([], out, assert_raises(RuntimeError, theano.function([], out,
mode=mode_local_assert)) mode=mode_local_assert))
finally: finally:
config.assert_no_cpu_op = old config.assert_no_cpu_op = old
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论