fix for quant model (#43567)

ae7192a8 · jakpiase · lidanqing-intel · e1604f9e · ae7192a8
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

paddle/fluid/operators/mkldnn/fc_mkldnn_op.cc paddle/fluid/operators/mkldnn/fc_mkldnn_op.cc +2 -0

未找到文件。
--- a/paddle/fluid/operators/mkldnn/fc_mkldnn_op.cc
+++ b/paddle/fluid/operators/mkldnn/fc_mkldnn_op.cc
@@ -178,6 +178,8 @@ class FCPrimitiveFactory {
    } else {
      out->set_format(in_format);
    }
+    out->set_mem_desc({phi::vectorize(out->dims()),
+                       platform::MKLDNNGetDataType<T_out>(), out->format()});
  }

  void UpdateDataPointers(const ExecutionContext& ctx, Tensor* out,