Lines Matching refs:parallelOp
208 LogicalResult matchAndRewrite(mlir::scf::ParallelOp parallelOp,
447 ParallelLowering::matchAndRewrite(ParallelOp parallelOp, in matchAndRewrite() argument
449 Location loc = parallelOp.getLoc(); in matchAndRewrite()
456 SmallVector<Value, 4> iterArgs = llvm::to_vector<4>(parallelOp.getInitVals()); in matchAndRewrite()
458 ivs.reserve(parallelOp.getNumLoops()); in matchAndRewrite()
462 llvm::zip(parallelOp.getInductionVars(), parallelOp.getLowerBound(), in matchAndRewrite()
463 parallelOp.getUpperBound(), parallelOp.getStep())) { in matchAndRewrite()
488 yieldOperands.reserve(parallelOp.getNumResults()); in matchAndRewrite()
489 for (auto &op : *parallelOp.getBody()) { in matchAndRewrite()
503 rewriter.eraseOp(parallelOp.getBody()->getTerminator()); in matchAndRewrite()
506 rewriter.mergeBlocks(parallelOp.getBody(), newBody, ivs); in matchAndRewrite()
508 rewriter.mergeBlockBefore(parallelOp.getBody(), newBody->getTerminator(), in matchAndRewrite()
518 rewriter.replaceOp(parallelOp, loopResults); in matchAndRewrite()