From 13ad8bde2a1b973e545e99c334edd4eb27f9915c Mon Sep 17 00:00:00 2001
From: jakpiase <jakpia21@gmail.com>
Date: Thu, 16 Jun 2022 09:12:33 +0200
Subject: [PATCH] fix for quant model (#43567)

---
 paddle/fluid/operators/mkldnn/fc_mkldnn_op.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/paddle/fluid/operators/mkldnn/fc_mkldnn_op.cc b/paddle/fluid/operators/mkldnn/fc_mkldnn_op.cc
index 590ffe4d0d4..8621e79509a 100644
--- 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,
-- 
GitLab