提交 7b362c83 authored 作者: Frederic's avatar Frederic

Enable AbstractOpt cudnn opt and crash fix in it.

上级 18ef99c3
...@@ -2448,9 +2448,9 @@ def local_abstractconv_cudnn(node): ...@@ -2448,9 +2448,9 @@ def local_abstractconv_cudnn(node):
inp1 = node.inputs[0] inp1 = node.inputs[0]
inp2 = node.inputs[1] inp2 = node.inputs[1]
if ((not isinstance(node.op, AbstractConv2d) or if (not isinstance(node.op, (AbstractConv2d,
not isinstance(node.op, AbstractConv2d_gradWeights) or AbstractConv2d_gradWeights,
not isinstance(node.op, AbstractConv2d_gradInputs))): AbstractConv2d_gradInputs))):
return None return None
if not isinstance(inp1.type, CudaNdarrayType) or \ if not isinstance(inp1.type, CudaNdarrayType) or \
...@@ -2459,7 +2459,7 @@ def local_abstractconv_cudnn(node): ...@@ -2459,7 +2459,7 @@ def local_abstractconv_cudnn(node):
if not dnn_available(): if not dnn_available():
return None return None
if node.op.filters_flip: if node.op.filter_flip:
conv_mode = 'conv' conv_mode = 'conv'
else: else:
conv_mode = 'cross' conv_mode = 'cross'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论