Lines Matching refs:parallelOp
335 LogicalResult matchAndRewrite(scf::ParallelOp parallelOp, in matchAndRewrite()
341 for (auto reduce : parallelOp.getOps<scf::ReduceOp>()) { in matchAndRewrite()
351 Location loc = parallelOp.getLoc(); in matchAndRewrite()
355 reductionVariables.reserve(parallelOp.getNumReductions()); in matchAndRewrite()
356 for (Value init : parallelOp.getInitVals()) { in matchAndRewrite()
371 llvm::zip(parallelOp.getOps<scf::ReduceOp>(), reductionVariables)) { in matchAndRewrite()
390 parallelOp.getLoc(), parallelOp.getLowerBound(), in matchAndRewrite()
391 parallelOp.getUpperBound(), parallelOp.getStep()); in matchAndRewrite()
394 rewriter.inlineRegionBefore(parallelOp.getRegion(), loop.region(), in matchAndRewrite()
402 auto scope = rewriter.create<memref::AllocaScopeOp>(parallelOp.getLoc(), in matchAndRewrite()
426 rewriter.replaceOp(parallelOp, results); in matchAndRewrite()