提交 676938f9 authored 作者: Frederic's avatar Frederic

fix test when cudnn is available

上级 b5cca42e
...@@ -174,15 +174,16 @@ def test_gpuspecifyshape(): ...@@ -174,15 +174,16 @@ def test_gpuspecifyshape():
assert not numpy.any([isinstance(x.op, cuda.HostFromGpu) for x in l]) assert not numpy.any([isinstance(x.op, cuda.HostFromGpu) for x in l])
def test_softmax(): def test_softmax():
x = tensor.fmatrix() x = tensor.fmatrix()
f = theano.function([x],tensor.nnet.nnet.Softmax()(x), mode=mode_with_gpu) f = theano.function([x], tensor.nnet.nnet.Softmax()(x),
f2 = theano.function([x],tensor.nnet.nnet.Softmax()(x), mode=mode_without_gpu) mode=mode_with_gpu.excluding('cudnn'))
assert isinstance(f.maker.fgraph.toposort()[1].op,cuda.nnet.GpuSoftmax) f2 = theano.function([x], tensor.nnet.nnet.Softmax()(x),
xv=numpy.random.rand(7,8).astype('float32') mode=mode_without_gpu)
assert numpy.allclose(f(xv),f2(xv)) assert isinstance(f.maker.fgraph.toposort()[1].op, cuda.nnet.GpuSoftmax)
xv = numpy.random.rand(7, 8).astype('float32')
assert numpy.allclose(f(xv), f2(xv))
def test_softmax_with_bias(): def test_softmax_with_bias():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论