From 590b4dbcdd989324089ce43c22ef151c746c92a3 Mon Sep 17 00:00:00 2001
From: xiongkun <xiongkun03@baidu.com>
Date: Thu, 5 May 2022 20:30:39 +0800
Subject: [PATCH] fix bugs (#42495)

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

diff --git a/paddle/phi/infermeta/unary.cc b/paddle/phi/infermeta/unary.cc
index 47f505e53f4..b7959d2809e 100644
--- a/paddle/phi/infermeta/unary.cc
+++ b/paddle/phi/infermeta/unary.cc
@@ -436,6 +436,8 @@ void EinsumInferMeta(const std::vector<const MetaTensor*>& inputs,
           << paddle::string::join_strings(output_dims, ",");
   VLOG(3) << "Label Type is : " << label_to_string(all_labels, labeltype);
   VLOG(3) << "Label Shape is : " << label_to_string(all_labels, labelshape);
+  out->set_dims(make_ddim(output_dims));
+  out->set_dtype(inputs[0]->dtype());
 }
 
 void ExpandInferMeta(const MetaTensor& x,
-- 
GitLab