fix bugs (#42495)

2742195759 · web-flow · commit 590b4dbcdd98 · 2022-05-05T20:30:39.000+08:00
diff --git 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,

Original file line number	Diff line number	Diff line change
`@@ -436,6 +436,8 @@ void EinsumInferMeta(const std::vector<const MetaTensor*>& inputs,`
`436`	`436`	`<< paddle::string::join_strings(output_dims, ",");`
`437`	`437`	`VLOG(3) << "Label Type is : " << label_to_string(all_labels, labeltype);`
`438`	`438`	`VLOG(3) << "Label Shape is : " << label_to_string(all_labels, labelshape);`
	`439`	`+ out->set_dims(make_ddim(output_dims));`
	`440`	`+ out->set_dtype(inputs[0]->dtype());`
`439`	`441`	`}`
`440`	`442`
`441`	`443`	`void ExpandInferMeta(const MetaTensor& x,`