deconv small fix

7 years ago · 0efac253d3
parent 2d956b82cd
commit 0efac253d3
1 changed files with 1 additions and 0 deletions
--- a/paddle/operators/conv2d_transpose_cudnn_op.cu
+++ b/paddle/operators/conv2d_transpose_cudnn_op.cu
@ -43,6 +43,7 @@ class CudnnConvTransposeOpKernel : public framework::OpKernel<T> {

    std::vector<int> strides = ctx.Attr<std::vector<int>>("strides");
    std::vector<int> paddings = ctx.Attr<std::vector<int>>("paddings");
+    // cudnn v5 does not support dilations
    std::vector<int> dilations = ctx.Attr<std::vector<int>>("dilations");
    int user_workspace_size = ctx.Attr<int>("workspace_size_MB");