fix nullptr bug of BmmGradInferMeta (#45765)

26d161ef · BiynXu · GitHub · 22255528 · 26d161ef
隐藏空白更改
内联并排

Showing with 8 addition and 4 deletion

paddle/phi/infermeta/backward.cc paddle/phi/infermeta/backward.cc +8 -4

未找到文件。
--- a/paddle/phi/infermeta/backward.cc
+++ b/paddle/phi/infermeta/backward.cc
@@ -88,10 +88,14 @@ void BmmGradInferMeta(const MetaTensor& x,
                      const MetaTensor& out_grad,
                      MetaTensor* x_grad,
                      MetaTensor* y_grad) {
-  x_grad->set_dims(x.dims());
+  if (x_grad) {
-  y_grad->set_dims(y.dims());
+    x_grad->set_dims(x.dims());
-  x_grad->set_dtype(x.dtype());
+    x_grad->set_dtype(x.dtype());
-  y_grad->set_dtype(y.dtype());
+  }
+  if (y_grad) {
+    y_grad->set_dims(y.dims());
+    y_grad->set_dtype(y.dtype());
+  }
 }
 void ChannelShuffleGradInferMeta(const MetaTensor& out_grad,