提交 4f2a0f22 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Use mode=m for functions built in test_hostfromgpu_shape_i.

This makes the test pass with FAST_COMPILE
上级 5cfa6e2b
...@@ -450,9 +450,9 @@ def test_hostfromgpu_shape_i(): ...@@ -450,9 +450,9 @@ def test_hostfromgpu_shape_i():
av=numpy.asarray(numpy.random.rand(5,4),dtype='float32') av=numpy.asarray(numpy.random.rand(5,4),dtype='float32')
cv=cuda.CudaNdarray(numpy.asarray(numpy.random.rand(5,4),dtype='float32')) cv=cuda.CudaNdarray(numpy.asarray(numpy.random.rand(5,4),dtype='float32'))
f = theano.function([a],cuda.basic_ops.gpu_from_host(a)) f = theano.function([a],cuda.basic_ops.gpu_from_host(a), mode=m)
assert cuda.basic_ops.gpu_from_host in [x.op for x in f.maker.env.toposort()] assert cuda.basic_ops.gpu_from_host in [x.op for x in f.maker.env.toposort()]
f = theano.function([a],cuda.basic_ops.gpu_from_host(a).shape) f = theano.function([a],cuda.basic_ops.gpu_from_host(a).shape, mode=m)
topo = f.maker.env.toposort() topo = f.maker.env.toposort()
assert isinstance(topo[0].op,T.opt.Shape_i) assert isinstance(topo[0].op,T.opt.Shape_i)
assert isinstance(topo[1].op,T.opt.Shape_i) assert isinstance(topo[1].op,T.opt.Shape_i)
...@@ -461,9 +461,9 @@ def test_hostfromgpu_shape_i(): ...@@ -461,9 +461,9 @@ def test_hostfromgpu_shape_i():
f = theano.function([ca],cuda.basic_ops.host_from_gpu(ca)) f = theano.function([ca],cuda.basic_ops.host_from_gpu(ca), mode=m)
assert cuda.basic_ops.host_from_gpu in [x.op for x in f.maker.env.toposort()] assert cuda.basic_ops.host_from_gpu in [x.op for x in f.maker.env.toposort()]
f = theano.function([ca],cuda.basic_ops.host_from_gpu(ca).shape) f = theano.function([ca],cuda.basic_ops.host_from_gpu(ca).shape, mode=m)
topo = f.maker.env.toposort() topo = f.maker.env.toposort()
assert isinstance(topo[0].op,T.opt.Shape_i) assert isinstance(topo[0].op,T.opt.Shape_i)
assert isinstance(topo[1].op,T.opt.Shape_i) assert isinstance(topo[1].op,T.opt.Shape_i)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论