Searched refs:parOp (Results 1 – 3 of 3) sorted by relevance
164 if (scf::ParallelOp parOp = scf::getParallelForInductionVarOwner(iv)) { in matchAndRewrite() local165 for (unsigned idx = 0; idx < parOp.getNumLoops(); ++idx) { in matchAndRewrite()166 if (parOp.getInductionVars()[idx] == iv) { in matchAndRewrite()167 lb = parOp.getLowerBound()[idx]; in matchAndRewrite()168 ub = parOp.getUpperBound()[idx]; in matchAndRewrite()169 step = parOp.getStep()[idx]; in matchAndRewrite()
202 scf::ParallelOp parOp; in matchAndRewrite() local205 parOp = rewriter.create<scf::ParallelOp>(loc, lowerBoundTuple, in matchAndRewrite()208 rewriter.eraseBlock(parOp.getBody()); in matchAndRewrite()210 parOp.getRegion().end()); in matchAndRewrite()211 rewriter.replaceOp(op, parOp.getResults()); in matchAndRewrite()231 parOp = rewriter.create<scf::ParallelOp>( in matchAndRewrite()236 rewriter.eraseBlock(parOp.getBody()); in matchAndRewrite()237 rewriter.inlineRegionBefore(op.getRegion(), parOp.getRegion(), in matchAndRewrite()238 parOp.getRegion().end()); in matchAndRewrite()248 rewriter.setInsertionPoint(&parOp.getBody()->back()); in matchAndRewrite()[all …]
1223 scf::ParallelOp parOp = builder.create<scf::ParallelOp>(loc, lo, hi, step); in genFor() local1225 codegen.pidxs[tensor][idx] = parOp.getInductionVars()[0]; in genFor()1227 codegen.loops[idx] = parOp.getInductionVars()[0]; in genFor()1228 builder.setInsertionPointToStart(parOp.getBody()); in genFor()1229 return parOp; in genFor()