提交 a341b215 authored 作者: Caglar's avatar Caglar

fixed the mode in test.

上级 0c6cfb37
......@@ -96,7 +96,7 @@ def test_local_assert_no_cpu_op():
m = np.random.uniform(-1, 1, (10, 10)).astype("float32")
ms = theano.shared(m, name="m_shared")
z = ms**2 + 3
mode_local_assert = mode.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")
f = theano.function([], z, mode=mode_local_assert)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论