[Mlir-commits] [mlir] [MLIR] Add `InParallelOpInterface` for parallel combining operations (PR #157736)
Mehdi Amini
llvmlistbot at llvm.org
Tue Sep 9 12:52:27 PDT 2025
================
@@ -1911,8 +1917,9 @@ struct FoldTensorCastOfOutputIntoForallOp
auto terminator = newForallOp.getTerminator();
for (auto [yieldingOp, outputBlockArg] : llvm::zip(
terminator.getYieldingOps(), newForallOp.getRegionIterArgs())) {
- auto insertSliceOp = cast<tensor::ParallelInsertSliceOp>(yieldingOp);
- insertSliceOp.getDestMutable().assign(outputBlockArg);
+ auto insertSliceOp = dyn_cast<tensor::ParallelInsertSliceOp>(yieldingOp);
+ if (insertSliceOp)
+ insertSliceOp.getDestMutable().assign(outputBlockArg);
----------------
joker-eph wrote:
What happens when we hit something else than a `ParallelInsertSliceOp` here?
https://github.com/llvm/llvm-project/pull/157736
More information about the Mlir-commits
mailing list