Fix

co63oc · co63oc · commit 70a18ae01d43 · 2025-06-04T16:55:30.000+08:00
diff --git a/paddle/phi/infermeta/binary.cc b/paddle/phi/infermeta/binary.cc
@@ -2969,7 +2969,12 @@ void MatmulInferMeta(const MetaTensor& x,
   } else {
     new_dims.reserve(ndims_x);
     for (size_t i = 0; i < ndims_x - 2; ++i) {
-      new_dims.push_back(std::max(dims_x[i], dims_y[i]));
+      // If one of them is 0, choose 0.
+      if (dims_x[i] == 0 || dims_y[i] == 0) {
+        new_dims.push_back(0);
+      } else {
+        new_dims.push_back(std::max(dims_x[i], dims_y[i]));
+      }
     }
   }
   if (!x_broadcasted) {

Original file line number	Diff line number	Diff line change
`@@ -2969,7 +2969,12 @@ void MatmulInferMeta(const MetaTensor& x,`
`2969`	`2969`	`} else {`
`2970`	`2970`	`new_dims.reserve(ndims_x);`
`2971`	`2971`	`for (size_t i = 0; i < ndims_x - 2; ++i) {`
`2972`		`- new_dims.push_back(std::max(dims_x[i], dims_y[i]));`
	`2972`	`+ // If one of them is 0, choose 0.`
	`2973`	`+ if (dims_x[i] == 0 \|\| dims_y[i] == 0) {`
	`2974`	`+ new_dims.push_back(0);`
	`2975`	`+ } else {`
	`2976`	`+ new_dims.push_back(std::max(dims_x[i], dims_y[i]));`
	`2977`	`+ }`
`2973`	`2978`	`}`
`2974`	`2979`	`}`
`2975`	`2980`	`if (!x_broadcasted) {`