提交 df59a925 authored 作者: f0k's avatar f0k

Allow alternative cuDNN ops in convolution tests

上级 cf351646
......@@ -594,7 +594,7 @@ def test_gemm_valid():
def test_dnn_valid():
if not cuda.dnn.dnn_available():
raise SkipTest(cuda.dnn.dnn_available.msg)
for t in _test_valid(GpuDnnConv, mode=theano_mode.including("cudnn")):
for t in _test_valid(GpuDnnConvBase, mode=theano_mode.including("cudnn")):
yield t
......@@ -708,7 +708,7 @@ def test_gemm_full():
def test_dnn_full():
if not cuda.dnn.dnn_available():
raise SkipTest(cuda.dnn.dnn_available.msg)
for t in _test_full(GpuDnnConv, mode=theano_mode.including("cudnn")):
for t in _test_full(GpuDnnConvBase, mode=theano_mode.including("cudnn")):
yield t
......@@ -760,7 +760,7 @@ def test_gemm_subsample():
def test_dnn_subsample():
if not cuda.dnn.dnn_available():
raise SkipTest(cuda.dnn.dnn_available.msg)
for t in _test_subsample(GpuDnnConv, theano_mode.including('cudnn')):
for t in _test_subsample(GpuDnnConvBase, theano_mode.including('cudnn')):
yield t
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论