add set_dtype for inverse_op (#44618)

PaddlePaddle · Jul 27, 2022 · ce7c799 · ce7c799
1 parent a71cfd8
commit ce7c799
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -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);
 }
-Original file line number
+Diff line change
@@ Expand Up @@
       }
       out->set_dims(input_dims);
+      out->set_dtype(x.dtype());
       out->share_lod(x);
     }
@@ Expand Down @@