Lines Matching refs:parallelOp
200 LogicalResult matchAndRewrite(mlir::scf::ParallelOp parallelOp,
404 ParallelLowering::matchAndRewrite(ParallelOp parallelOp, in matchAndRewrite() argument
406 Location loc = parallelOp.getLoc(); in matchAndRewrite()
413 SmallVector<Value, 4> iterArgs = llvm::to_vector<4>(parallelOp.initVals()); in matchAndRewrite()
415 ivs.reserve(parallelOp.getNumLoops()); in matchAndRewrite()
419 llvm::zip(parallelOp.getInductionVars(), parallelOp.lowerBound(), in matchAndRewrite()
420 parallelOp.upperBound(), parallelOp.step())) { in matchAndRewrite()
445 yieldOperands.reserve(parallelOp.getNumResults()); in matchAndRewrite()
446 for (auto &op : *parallelOp.getBody()) { in matchAndRewrite()
460 rewriter.eraseOp(parallelOp.getBody()->getTerminator()); in matchAndRewrite()
463 rewriter.mergeBlocks(parallelOp.getBody(), newBody, ivs); in matchAndRewrite()
465 rewriter.mergeBlockBefore(parallelOp.getBody(), newBody->getTerminator(), in matchAndRewrite()
475 rewriter.replaceOp(parallelOp, loopResults); in matchAndRewrite()