From fcbcfe44cff00101a6a98a73971398eb8dd87710 Mon Sep 17 00:00:00 2001 From: Maksim Levental Date: Fri, 25 Jul 2025 13:50:15 -0500 Subject: [mlir][NFC] update `mlir/Dialect` create APIs (32/n) (#150657) See https://github.com/llvm/llvm-project/pull/147168 for more info. --- .../VectorTransferSplitRewritePatterns.cpp | 35 +++++++++++----------- 1 file changed, 17 insertions(+), 18 deletions(-) (limited to 'mlir/lib/Dialect/Vector/Transforms/VectorTransferSplitRewritePatterns.cpp') diff --git a/mlir/lib/Dialect/Vector/Transforms/VectorTransferSplitRewritePatterns.cpp b/mlir/lib/Dialect/Vector/Transforms/VectorTransferSplitRewritePatterns.cpp index 05b00744beea..5e12dc486e59 100644 --- a/mlir/lib/Dialect/Vector/Transforms/VectorTransferSplitRewritePatterns.cpp +++ b/mlir/lib/Dialect/Vector/Transforms/VectorTransferSplitRewritePatterns.cpp @@ -348,24 +348,23 @@ getLocationToWriteFullVec(RewriterBase &b, vector::TransferWriteOp xferOp, Location loc = xferOp.getLoc(); Value zero = arith::ConstantIndexOp::create(b, loc, 0); Value memref = xferOp.getBase(); - return b - .create( - loc, inBoundsCond, - [&](OpBuilder &b, Location loc) { - Value res = - castToCompatibleMemRefType(b, memref, compatibleMemRefType); - scf::ValueVector viewAndIndices{res}; - llvm::append_range(viewAndIndices, xferOp.getIndices()); - scf::YieldOp::create(b, loc, viewAndIndices); - }, - [&](OpBuilder &b, Location loc) { - Value casted = - castToCompatibleMemRefType(b, alloc, compatibleMemRefType); - scf::ValueVector viewAndIndices{casted}; - viewAndIndices.insert(viewAndIndices.end(), - xferOp.getTransferRank(), zero); - scf::YieldOp::create(b, loc, viewAndIndices); - }) + return scf::IfOp::create( + b, loc, inBoundsCond, + [&](OpBuilder &b, Location loc) { + Value res = + castToCompatibleMemRefType(b, memref, compatibleMemRefType); + scf::ValueVector viewAndIndices{res}; + llvm::append_range(viewAndIndices, xferOp.getIndices()); + scf::YieldOp::create(b, loc, viewAndIndices); + }, + [&](OpBuilder &b, Location loc) { + Value casted = + castToCompatibleMemRefType(b, alloc, compatibleMemRefType); + scf::ValueVector viewAndIndices{casted}; + viewAndIndices.insert(viewAndIndices.end(), + xferOp.getTransferRank(), zero); + scf::YieldOp::create(b, loc, viewAndIndices); + }) ->getResults(); } -- cgit v1.2.3