[Mlir-commits] [mlir] [MLIR][mesh] Mesh fixes (PR #124724)

[Frank Schlimbach]

================
@@ -796,8 +790,12 @@ spmdizeFuncOp(FunctionOpInterface op, IRMapping &spmdizationMap,
   // Snapshot the original blocks to not mess up the iteration when adding new
   // blocks.
   SmallVector<Block *> originalBlocks;
-  llvm::transform(op.getBlocks(), std::back_inserter(originalBlocks),
-                  [](Block &b) { return &b; });
+  for (Block &b : op.getBlocks()) {
----------------
fschlimb wrote:

Without `ShardOp`s there is nothing to spmdize. The spmdize machinery however expects fully annotated shardings. 
Non-annotated blocks are expected to operate scalar-like.



https://github.com/llvm/llvm-project/pull/124724