Skip to content

deconv cudnn forward passed#5235

Merged
zchen0211 merged 22 commits intoPaddlePaddle:developfrom
zchen0211:develop
Nov 2, 2017
Merged

deconv cudnn forward passed#5235
zchen0211 merged 22 commits intoPaddlePaddle:developfrom
zchen0211:develop

Conversation

@zchen0211
Copy link
Contributor

working on backward now...

@chengduoZH
Copy link
Contributor

You should remove uncorrelated commits. Maybe git cherry-pick is helpful to you.

} // namespace paddle

namespace ops = paddle::operators;
REGISTER_OP(conv2dtranspose_cudnn, ops::Conv2DTransposeOp,
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is it appropriate that replacing conv2dtranspose_cudnn with conv2d_transpose_cudnn

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done.

@chengduoZH
Copy link
Contributor

deconv cudnn and conv cudnn have a lot of common parts. I think we can write these together.

using DataLayout = platform::DataLayout;
using CUDADeviceContext = platform::CUDADeviceContext;

static constexpr size_t kCONV_CUDNN_WORKSPACE_LIMIT_BYTES = 1024 * 1024 * 1024;
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.


std::vector<int> strides = ctx.Attr<std::vector<int>>("strides");
std::vector<int> paddings = ctx.Attr<std::vector<int>>("paddings");
std::vector<int> dilations = ctx.Attr<std::vector<int>>("dilations");
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

add a comment that cuDNN does not support dilations.


namespace ops = paddle::operators;

REGISTER_OP_GPU_KERNEL(conv2d_transpose_cudnn,
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

why name it conv2d_transpose?

Copy link
Member

@jacquesqiao jacquesqiao left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@zchen0211 zchen0211 merged commit 6d649d9 into PaddlePaddle:develop Nov 2, 2017
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

9 participants