提交 351ec3fb authored 作者: Frederic Bastien's avatar Frederic Bastien

fix test in fast_compile.

上级 5b0f3a16
......@@ -35,6 +35,7 @@ def test_shape_i():
v = cuda.CudaNdarray(numpy.zeros((3,4,5),dtype='float32'))
f = theano.function([x],x.shape[1])
topo = f.maker.env.toposort()
assert len(topo)==1
assert isinstance(topo[0].op,T.opt.Shape_i)
assert f(v)==4
if theano.config.mode!='FAST_COMPILE':
assert len(topo)==1
assert isinstance(topo[0].op,T.opt.Shape_i)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论