【PIR】Change layout of binary infermeta (#72213)

liuruyan · web-flow · commit 09ef2751a81d · 2025-04-14T11:13:15.000+08:00
* change layout of binary infermeta

* fix ci
diff --git a/paddle/phi/infermeta/binary.cc b/paddle/phi/infermeta/binary.cc
@@ -1740,7 +1740,15 @@ void ElementwiseRawInferMeta(const MetaTensor& x,
     promote_result = x.dtype();
   }
   out->set_dtype(promote_result);
-  out->set_layout(x.layout());
+
+  // layout need change when meet input layout contain kNHWC
+  auto layout = [&]() {
+    if (x.layout() == DataLayout::kNHWC || y.layout() == DataLayout::kNHWC)
+      return DataLayout::kNHWC;
+    return x.layout();
+  }();
+
+  out->set_layout(layout);
   out->share_lod(x);
 }
 
