Home
last modified time | relevance | path

Searched refs:innerWarp (Results 1 – 1 of 1) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDistribute.cpp811 auto innerWarp = rewriter.create<WarpExecuteOnLane0Op>( in matchAndRewrite() local
818 for (Value args : innerWarp.getBody()->getArguments()) { in matchAndRewrite()
825 rewriter.mergeBlocks(forOp.getBody(), innerWarp.getBody(), argMapping); in matchAndRewrite()
826 rewriter.setInsertionPoint(innerWarp.getBody(), innerWarp.getBody()->end()); in matchAndRewrite()
827 rewriter.create<vector::YieldOp>(innerWarp.getLoc(), yieldOperands); in matchAndRewrite()
828 rewriter.setInsertionPointAfter(innerWarp); in matchAndRewrite()
829 if (!innerWarp.getResults().empty()) in matchAndRewrite()
830 rewriter.create<scf::YieldOp>(forOp.getLoc(), innerWarp.getResults()); in matchAndRewrite()