提交 4ba23e7d authored 作者: Caglar's avatar Caglar

fixed test to register the optimization.

上级 26d43bf0
......@@ -98,8 +98,9 @@ def test_local_assert_no_cpu_op():
ms = cuda.shared_constructor(m, name="m_shared")
out = theano.tensor.tanh(ms**2 + 3)
mode_local_assert = mode_with_gpu.including("local_assert_no_cpu_op")
mode_local_assert = mode_with_gpu.including("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
......@@ -113,6 +114,7 @@ def test_local_assert_no_cpu_op():
# If the flag is ignore
try:
config.assert_no_cpu_op = 'ignore'
theano.function([], out, mode=mode_local_assert)
finally:
config.assert_no_cpu_op = old
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论