提交 b4bd7b90 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

No error in v3 when padding is of 0

上级 74b59451
...@@ -1401,7 +1401,8 @@ class GpuDnnPoolDesc(GpuOp): ...@@ -1401,7 +1401,8 @@ class GpuDnnPoolDesc(GpuOp):
raise RuntimeError("CuDNN pooling with padding requires CuDNN v2") raise RuntimeError("CuDNN pooling with padding requires CuDNN v2")
if self.get_ndim() == 3 and version() < (3000, 3000): if self.get_ndim() == 3 and version() < (3000, 3000):
raise RuntimeError("CuDNN 3d pooling requires CuDNN v3") raise RuntimeError("CuDNN 3d pooling requires CuDNN v3")
if mode == 'average_exc_pad' and version() < (4004, 4004): if (mode == 'average_exc_pad' and max(pad) > 0 and
version() < (4004, 4004)):
raise RuntimeError( raise RuntimeError(
"CuDNN pooling mode 'average_exc_pad' requires at least v4") "CuDNN pooling mode 'average_exc_pad' requires at least v4")
......
...@@ -953,7 +953,7 @@ class GpuDnnPoolDesc(Op): ...@@ -953,7 +953,7 @@ class GpuDnnPoolDesc(Op):
if self.get_ndim() == 3 and version() < 3000: if self.get_ndim() == 3 and version() < 3000:
raise RuntimeError("CuDNN 3d pooling requires v3") raise RuntimeError("CuDNN 3d pooling requires v3")
if mode == 'average_exc_pad' and version() < 4004: if mode == 'average_exc_pad' and max(pad) > 0 and version() < 4004:
raise RuntimeError( raise RuntimeError(
"CuDNN pooling mode 'average_exc_pad' requires at least v4") "CuDNN pooling mode 'average_exc_pad' requires at least v4")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论