From 112e3edbf64e5326eba9ad26c049f816ef278783 Mon Sep 17 00:00:00 2001 From: wangchaochaohu Date: Mon, 16 Mar 2020 21:32:42 +0800 Subject: [PATCH] fix the conv group problem test=develop (#23025) --- paddle/fluid/operators/conv_cudnn_op.cu | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/operators/conv_cudnn_op.cu b/paddle/fluid/operators/conv_cudnn_op.cu index cc5ae4c837..c885bf0d50 100644 --- 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 { cudnnConvolutionBwdFilterAlgo_t filter_algo = static_cast(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; -- GitLab