diff options
| author | Maksim Levental <maksim.levental@gmail.com> | 2025-07-25 13:50:15 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-07-25 13:50:15 -0500 |
| commit | fcbcfe44cff00101a6a98a73971398eb8dd87710 (patch) | |
| tree | 27aefe495459cd877856dc2b6300a11128342980 /mlir/lib/Dialect/Vector/Transforms/VectorTransferSplitRewritePatterns.cpp | |
| parent | c99b1bcd505064f2e086e6b1034ce0b0c91ea5b9 (diff) | |
[mlir][NFC] update `mlir/Dialect` create APIs (32/n) (#150657)
See https://github.com/llvm/llvm-project/pull/147168 for more info.
Diffstat (limited to 'mlir/lib/Dialect/Vector/Transforms/VectorTransferSplitRewritePatterns.cpp')
| -rw-r--r-- | mlir/lib/Dialect/Vector/Transforms/VectorTransferSplitRewritePatterns.cpp | 35 |
1 files changed, 17 insertions, 18 deletions
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<scf::IfOp>( - 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(); } |
