diff --git a/paddle/operators/conv2d_transpose_cudnn_op.cu b/paddle/operators/conv2d_transpose_cudnn_op.cu index 5a286897e06ae09753fb181d6482cd583430774b..61fcfb3bd8fa57f2c45fbf3a980dbe41041cff18 100644 --- 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 { std::vector strides = ctx.Attr>("strides"); std::vector paddings = ctx.Attr>("paddings"); + // cudnn v5 does not support dilations std::vector dilations = ctx.Attr>("dilations"); int user_workspace_size = ctx.Attr("workspace_size_MB");