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

Fix test. use float32 and enable the GPU.

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