提交 0c6cfb37 authored 作者: Caglar's avatar Caglar

added the test.

上级 d4fcb50a
......@@ -92,10 +92,14 @@ def test_local_gpu_contiguous_gpu_contiguous():
def test_local_assert_no_cpu_op():
x = theano.tensor.fscalar()
y = theano.tensor.fscalar()
z = x * y
f = theano.function([x, y], z, mode=mode_with_gpu)
numpy.random.seed(1)
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_local_assert.excluding("local_gpu_elemwise_1")
f = theano.function([], z, mode=mode_local_assert)
topo = f.maker.fgraph.toposort()
a_op = [n for n in topo if not isinstance(n.op, cuda.GpuElemwise)]
assert len(a_op) == 3
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论