提交 44a208cf authored 作者: Frederic's avatar Frederic

Make a test do what it should.

It was passing event if it tested the wrong stuff. After fixing it still work correctly.
上级 495f6cba
......@@ -68,9 +68,9 @@ def test_gpualloc():
x = theano.shared(numpy.ones(3, dtype='float32'), 'x')
m = (x).dimshuffle(['x', 0])
v = tensor.alloc(1., *m.shape)
f = theano.function([], v+x)
f = theano.function([], v + x, mode=mode_with_gpu)
l = f.maker.fgraph.toposort()
assert numpy.any(ininstance(x.op, cuda.GpuAlloc) for x in l )
assert numpy.any([isinstance(x.op, cuda.GpuAlloc) for x in l])
def test_alloc_memset_0():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论