提交 d7e5b2d6 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix GPU tests

上级 619d05f1
......@@ -92,7 +92,8 @@ def test_flatten():
assert GpuReshape in [type(node.op)
for node in f.maker.fgraph.toposort()]
f = theano.function([m], m.flatten(ndim=2), mode=mode_with_gpu)
f = theano.function([m], m.flatten(ndim=2),
mode=mode_with_gpu.excluding("local_useless_reshape"))
val = np.random.rand(10, 11).astype("float32")
res = f(val)
utt.assert_allclose(res, val)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论