未验证 提交 112e3edb 编写于 作者: W wangchaochaohu 提交者: GitHub

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

上级 91b0a9ab
...@@ -559,7 +559,8 @@ class CUDNNConvGradOpKernel : public framework::OpKernel<T> { ...@@ -559,7 +559,8 @@ class CUDNNConvGradOpKernel : public framework::OpKernel<T> {
cudnnConvolutionBwdFilterAlgo_t filter_algo = cudnnConvolutionBwdFilterAlgo_t filter_algo =
static_cast<cudnnConvolutionBwdFilterAlgo_t>(0); static_cast<cudnnConvolutionBwdFilterAlgo_t>(0);
size_t workspace_size = 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) #if CUDNN_VERSION_MIN(7, 0, 1)
iwo_groups = 1; iwo_groups = 1;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册