提交 28d64c99 authored 作者: Frederic's avatar Frederic

Fix test. use float32 and enable the GPU.

上级 11b080f2
......@@ -72,21 +72,21 @@ def test_pooling():
def test_pooling_opt():
if not cuda.dnn.dnn_available():
raise SkipTest(cuda.dnn.dnn_available.msg)
x = T.tensor4()
f = theano.function([x],
x = T.ftensor4()
f = theano.function(
[x],
max_pool_2d(x, ds=(2, 2)),
mode=theano.compile.mode.get_mode('FAST_RUN').including("cudnn"))
mode=mode_with_gpu.including("cudnn"))
assert any([isinstance(n.op, cuda.dnn.GpuDnnPool)
for n in f.maker.fgraph.toposort()])
f = theano.function([x],
T.grad(max_pool_2d(x, ds=(2, 2)).sum(), x),
mode=theano.compile.mode.get_mode('FAST_RUN').including("cudnn"))
assert any([isinstance(n.op, cuda.dnn.GpuDnnPoolGrad)
for n in f.maker.fgraph.toposort()])
for n in f.maker.fgraph.toposort()])
f = theano.function(
[x],
T.grad(max_pool_2d(x, ds=(2, 2)).sum(), x),
mode=mode_with_gpu.including("cudnn"))
assert any([isinstance(n.op, cuda.dnn.GpuDnnPoolGrad)
for n in f.maker.fgraph.toposort()])
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论