fix the conv group problem test=develop (#23025)

5 years ago · 112e3edbf6
parent 91b0a9ab23
commit 112e3edbf6
1 changed files with 2 additions and 1 deletions
--- a/paddle/fluid/operators/conv_cudnn_op.cu
+++ b/paddle/fluid/operators/conv_cudnn_op.cu
@ -559,7 +559,8 @@ class CUDNNConvGradOpKernel : public framework::OpKernel<T> {
    cudnnConvolutionBwdFilterAlgo_t filter_algo =
        static_cast<cudnnConvolutionBwdFilterAlgo_t>(0);
    size_t workspace_size = 0;
-    int iwo_groups, c_groups;
+    int iwo_groups = groups;
+    int c_groups = 1;

 #if CUDNN_VERSION_MIN(7, 0, 1)
    iwo_groups = 1;