From ce7c79953ed093088bdc68f4e593d3f1fd3068bd Mon Sep 17 00:00:00 2001
From: freeliuzc <lzc842650834@gmail.com>
Date: Wed, 27 Jul 2022 10:46:43 +0800
Subject: [PATCH] add set_dtype for inverse_op (#44618)

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

diff --git a/paddle/phi/infermeta/backward.cc b/paddle/phi/infermeta/backward.cc
index 1eca092a5f..a33b9587c1 100644
--- a/paddle/phi/infermeta/backward.cc
+++ b/paddle/phi/infermeta/backward.cc
@@ -419,6 +419,7 @@ void InverseGradInferMeta(const MetaTensor& out,
                           MetaTensor* dx) {
   if (dx) {
     dx->set_dims(dout.dims());
+    dx->set_dtype(out.dtype());
   }
 }
 
diff --git a/paddle/phi/infermeta/unary.cc b/paddle/phi/infermeta/unary.cc
index edc455225e..c018e58a59 100644
--- a/paddle/phi/infermeta/unary.cc
+++ b/paddle/phi/infermeta/unary.cc
@@ -1059,6 +1059,7 @@ void InverseInferMeta(const MetaTensor& x, MetaTensor* out) {
   }
 
   out->set_dims(input_dims);
+  out->set_dtype(x.dtype());
   out->share_lod(x);
 }
 
-- 
GitLab