提交 83eb0431 authored 作者: Frederic's avatar Frederic

[TESTS] fix tests following change of default behavior

上级 f620cf5d
...@@ -574,7 +574,7 @@ def test_gemm_valid(): ...@@ -574,7 +574,7 @@ def test_gemm_valid():
extra_shapes += get_shapes2(scales_kern=(2, 2), kern_stride=(2, 2)) extra_shapes += get_shapes2(scales_kern=(2, 2), kern_stride=(2, 2))
for t in _test_valid(cuda.blas.BaseGpuCorrMM, for t in _test_valid(cuda.blas.BaseGpuCorrMM,
mode=theano_mode.including("conv_gemm"), mode=theano_mode.excluding("cudnn"),
extra_shapes=extra_shapes): extra_shapes=extra_shapes):
yield t yield t
...@@ -683,7 +683,7 @@ def test_full(): ...@@ -683,7 +683,7 @@ def test_full():
def test_gemm_full(): def test_gemm_full():
for t in _test_full(cuda.blas.BaseGpuCorrMM, for t in _test_full(cuda.blas.BaseGpuCorrMM,
mode=theano_mode.including("conv_gemm")): mode=theano_mode.excluding("cudnn")):
yield t yield t
...@@ -735,7 +735,7 @@ def test_subsample(): ...@@ -735,7 +735,7 @@ def test_subsample():
def test_gemm_subsample(): def test_gemm_subsample():
for t in _test_subsample(cuda.blas.BaseGpuCorrMM, for t in _test_subsample(cuda.blas.BaseGpuCorrMM,
theano_mode.including("conv_gemm")): theano_mode.excluding("cudnn")):
yield t yield t
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论