From 0efac253d340b22999407d387a4c2098cb5581c2 Mon Sep 17 00:00:00 2001
From: zchen0211 <chenzhuoyuan07@gmail.com>
Date: Wed, 1 Nov 2017 17:16:53 -0700
Subject: [PATCH] deconv small fix

---
 paddle/operators/conv2d_transpose_cudnn_op.cu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/paddle/operators/conv2d_transpose_cudnn_op.cu b/paddle/operators/conv2d_transpose_cudnn_op.cu
index 5a286897e06..61fcfb3bd8f 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<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");
 
-- 
GitLab