提交 8971b5ca authored 作者: abergeron's avatar abergeron

Merge pull request #2264 from nouiz/crash_fix_dnn

fix gh-2262. Fix crash cause by using cudnn when we shouldn't
......@@ -1164,6 +1164,8 @@ if cuda_available:
and (isinstance(node.inputs[0].owner.op, HostFromGpu)
or isinstance(node.inputs[1].owner.op, HostFromGpu))
):
if not dnn_available():
return
ins = []
for n in node.inputs:
if isinstance(n.owner.op, HostFromGpu):
......
......@@ -574,7 +574,7 @@ def test_gemm_valid():
extra_shapes += get_shapes2(scales_kern=(2, 2), kern_stride=(2, 2))
for t in _test_valid(cuda.blas.BaseGpuCorrMM,
mode=theano_mode.including("conv_gemm"),
mode=theano_mode.excluding("cudnn"),
extra_shapes=extra_shapes):
yield t
......@@ -683,7 +683,7 @@ def test_full():
def test_gemm_full():
for t in _test_full(cuda.blas.BaseGpuCorrMM,
mode=theano_mode.including("conv_gemm")):
mode=theano_mode.excluding("cudnn")):
yield t
......@@ -735,7 +735,7 @@ def test_subsample():
def test_gemm_subsample():
for t in _test_subsample(cuda.blas.BaseGpuCorrMM,
theano_mode.including("conv_gemm")):
theano_mode.excluding("cudnn")):
yield t
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论