提交 77fa4682 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix test following new optimization

上级 fa8e31d6
......@@ -39,7 +39,8 @@ def test_view():
a = rand_gpuarray(20, dtype=dtype)
g = GpuArrayType(dtype=dtype, broadcastable=(False,))('g')
f = theano.function([g], ViewOp()(g))
m = theano.compile.get_default_mode().excluding("local_view_op")
f = theano.function([g], ViewOp()(g), mode=m)
assert isinstance(f.maker.fgraph.toposort()[0].op, ViewOp)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论