提交 a1296c66 authored 作者: Kashif Rasul's avatar Kashif Rasul

fix shape mismatch in GpuDnnPoolGrad

fixes for issue #3347
上级 450dfcb2
...@@ -2321,10 +2321,10 @@ if True: ...@@ -2321,10 +2321,10 @@ if True:
(inp_grad.owner and isinstance(inp_grad.owner.op, (inp_grad.owner and isinstance(inp_grad.owner.op,
HostFromGpu))): HostFromGpu))):
desc = GpuDnnPoolDesc(ws=ds, stride=st, mode=mode, pad=pad)() desc = GpuDnnPoolDesc(ws=ds, stride=st, mode=mode, pad=pad)()
contiguous_inp_grad = gpu_contiguous(inp_grad)
ret = GpuDnnPoolGrad()(gpu_contiguous(inp), ret = GpuDnnPoolGrad()(gpu_contiguous(inp),
gpu_contiguous(numpy.empty((1,1,1,1), contiguous_inp_grad,
dtype=numpy.float32)), contiguous_inp_grad,
gpu_contiguous(inp_grad),
desc) desc)
return [host_from_gpu(ret)] return [host_from_gpu(ret)]
......
...@@ -1763,9 +1763,10 @@ def local_avg_pool_dnn_grad_stride(node): ...@@ -1763,9 +1763,10 @@ def local_avg_pool_dnn_grad_stride(node):
mode = node.op.mode mode = node.op.mode
desc = GpuDnnPoolDesc(ws=ds, stride=st, mode=mode, pad=pad)() desc = GpuDnnPoolDesc(ws=ds, stride=st, mode=mode, pad=pad)()
contiguous_inp_grad = gpu_contiguous(inp_grad)
return GpuDnnPoolGrad()(gpu_contiguous(inp), return GpuDnnPoolGrad()(gpu_contiguous(inp),
gpu_contiguous(numpy.empty((1, 1, 1, 1), dtype=numpy.float32)), contiguous_inp_grad,
gpu_contiguous(inp_grad), contiguous_inp_grad,
desc) desc)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论