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

fixed the assertion raised.

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