fix mul_mkldnn_op build failure (#18816)

6 years ago · c5f47c2107
parent a5c986301c
commit c5f47c2107
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/mkldnn/mul_mkldnn_op.cc
+++ b/paddle/fluid/operators/mkldnn/mul_mkldnn_op.cc
@ -84,7 +84,7 @@ class MulPrimitiveFactory {
    if ((data->dims().size() == 4 &&
         src_fmt != (dst_fmt = memory::format::nchw)) ||
        (data->dims().size() == 5 &&
-         dst_fmt != (dst_fmt = memory::format::ncdhw))) {
+         src_fmt != (dst_fmt = memory::format::ncdhw))) {
      auto dst_mdesc = CreateMemDescriptor<T>(data, dst_fmt);
      x_tmp.mutable_data<T>(ctx.GetPlace(), data->memory_size());