Set dtype for Infermeta

PaddlePaddle · Feb 25, 2022 · b7acb5d · b7acb5d · paddle-bot-old · Feb 25, 2022
1 parent a11094a
commit b7acb5d
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 2 deletions.
diff --git a/paddle/phi/infermeta/backward.cc b/paddle/phi/infermeta/backward.cc
@@ -48,15 +48,19 @@ void BilinearTensorProductGradInferMeta(const MetaTensor& x,
 
   if (dx) {
     dx->set_dims(x_dims);
+    dx->set_dtype(x.dtype());
   }
   if (dy) {
     dy->set_dims(y_dims);
+    dy->set_dtype(y.dtype());
   }
   if (dweight) {
     dweight->set_dims(weight_dims);
+    dweight->set_dtype(weight.dtype());
   }
   if (dbias) {
     dbias->set_dims({1, out_dims[1]});
+    dbias->set_dtype(dout.dtype());
   }
 }
 

diff --git a/paddle/phi/infermeta/multiary.cc b/paddle/phi/infermeta/multiary.cc
@@ -81,6 +81,7 @@ void BilinearTensorProductInferMeta(const MetaTensor& x,
 
   out->set_dims({x_dims[0], weight_dims[0]});
   out->share_lod(x);
+  out->set_dtype(x.dtype());
 }
 
 void ConcatInferMeta(const std::vector<MetaTensor>& x,

diff --git a/paddle/phi/kernels/bilinear_tensor_product_grad_kernel.h b/paddle/phi/kernels/bilinear_tensor_product_grad_kernel.h
@@ -15,7 +15,6 @@
 #pragma once
 
 #include "paddle/phi/core/dense_tensor.h"
-#include "paddle/utils/optional.h"
 
 namespace phi {
 

diff --git a/paddle/phi/kernels/impl/bilinear_tensor_product_grad_kernel_impl.h b/paddle/phi/kernels/impl/bilinear_tensor_product_grad_kernel_impl.h
@@ -17,7 +17,6 @@
 #include "paddle/phi/core/dense_tensor.h"
 #include "paddle/phi/kernels/funcs/blas/blas.h"
 #include "paddle/phi/kernels/funcs/eigen/common.h"
-#include "paddle/utils/optional.h"
 
 namespace phi {