Fix the weight md for group convolution

7b5ea706 · Haihao Shen · 59a90d08 · 7b5ea706
隐藏空白更改
内联并排

Showing with 1 addition and 2 deletion

paddle/fluid/operators/conv_mkldnn_op.cc paddle/fluid/operators/conv_mkldnn_op.cc +1 -2

未找到文件。
--- a/paddle/fluid/operators/conv_mkldnn_op.cc
+++ b/paddle/fluid/operators/conv_mkldnn_op.cc
@@ -566,8 +566,7 @@ class ConvMKLDNNOpKernel : public paddle::framework::OpKernel<T> {
            src_md.reset(new mkldnn::memory::desc(platform::MKLDNNMemDesc(
                src_tz, platform::MKLDNNGetDataType<float>(), chosen_memory_format)));
            weights_md.reset(new mkldnn::memory::desc(platform::MKLDNNMemDesc(
-                weights_tz, platform::MKLDNNGetDataType<float>(),
+                weights_tz, platform::MKLDNNGetDataType<float>(), chosen_memory_format)));
-                (g == 1) ? chosen_memory_format : mkldnn::memory::format::goihw)));
            dst_md.reset(new mkldnn::memory::desc(platform::MKLDNNMemDesc(
                dst_tz, platform::MKLDNNGetDataType<float>(), chosen_memory_format)));
            mds[2] = src_md;