From 74074a8d23574fb04d8d57b770a7a094050d0f63 Mon Sep 17 00:00:00 2001
From: zhupengyang <zhu_py@qq.com>
Date: Fri, 5 May 2023 09:54:36 +0800
Subject: [PATCH] fix xpu max size (#53505)

---
 paddle/phi/infermeta/fusion.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddle/phi/infermeta/fusion.cc b/paddle/phi/infermeta/fusion.cc
index f775cedce8c..fb7ede4c9e7 100644
--- a/paddle/phi/infermeta/fusion.cc
+++ b/paddle/phi/infermeta/fusion.cc
@@ -158,7 +158,7 @@ void Conv2dXPUInferMeta(const MetaTensor& x,
   }
   // set output and output max dims
   out->set_dims(DDim(out_shape.data(), out_shape.size()));
-  out_max->set_dims(phi::make_ddim({4}));
+  out_max->set_dims(phi::make_ddim({6}));
 }
 
 void EmbeddingWithEltwiseAddXPUInferMeta(
@@ -204,7 +204,7 @@ void FcXPUInferMeta(const MetaTensor& x,
   out->set_dims(DDim(out_shape.data(), out_shape.size()));
   out->set_dtype(x.dtype());
   out->set_layout(x.layout());
-  out_max->set_dims(w_max.dims());
+  out_max->set_dims(phi::make_ddim({6}));
   out_max->set_dtype(x.dtype());
   out_max->set_layout(x.layout());
 }
-- 
GitLab