提交 0719742c authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Allow fallback on GpuCorrMM if cuDNN is not there

上级 e679cd14
...@@ -2646,7 +2646,7 @@ def local_abstractconv_cudnn_graph(op, context_name, inputs, outputs): ...@@ -2646,7 +2646,7 @@ def local_abstractconv_cudnn_graph(op, context_name, inputs, outputs):
AbstractConv2d_gradInputs))): AbstractConv2d_gradInputs))):
return return
if version() < 6000 and op.filter_dilation != (1, 1): if version(raises=False) < 6000 and op.filter_dilation != (1, 1):
return None return None
inp1 = inputs[0] inp1 = inputs[0]
...@@ -2692,7 +2692,7 @@ def local_abstractconv3d_cudnn_graph(op, context_name, inputs, outputs): ...@@ -2692,7 +2692,7 @@ def local_abstractconv3d_cudnn_graph(op, context_name, inputs, outputs):
AbstractConv3d_gradInputs))): AbstractConv3d_gradInputs))):
return return
if version() < 6000 and op.filter_dilation != (1, 1, 1): if version(raises=False) < 6000 and op.filter_dilation != (1, 1, 1):
return None return None
inp1 = inputs[0] inp1 = inputs[0]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论