Lines Matching refs:newForOp
75 scf::ForOp newForOp,
243 auto newForOp = in createKernelLoop() local
246 return newForOp; in createKernelLoop()
250 scf::ForOp newForOp, in createKernel() argument
259 rewriter.setInsertionPoint(newForOp.getBody(), newForOp.getBody()->begin()); in createKernel()
261 mapping.map(forOp.getInductionVar(), newForOp.getInductionVar()); in createKernel()
263 mapping.map(arg.value(), newForOp.getRegionIterArgs()[arg.index()]); in createKernel()
270 newForOp.getLoc(), ub - (maxStage - i) * step); in createKernel()
272 newForOp.getLoc(), arith::CmpIPredicate::slt, in createKernel()
273 newForOp.getInductionVar(), c); in createKernel()
288 forOp.getLoc(), newForOp.getInductionVar(), offset); in createKernel()
323 newForOp.getRegionIterArgs()[remap->second]); in createKernel()
352 setValueMapping(it.first, newForOp->getResult(yieldOperands.size()), in createKernel()
355 newForOp.getBody()->getArguments()[yieldOperands.size() + 1 + in createKernel()
356 newForOp.getNumInductionVars()]); in createKernel()
358 setValueMapping(it.first, newForOp->getResult(yieldOperands.size()), in createKernel()
369 newForOp->getResult(retVal.index()), in createKernel()
464 ForOp newForOp = in returningMatchAndRewrite() local
468 pipeliner.createKernel(newForOp, crossStageValues, loopArgMap, rewriter); in returningMatchAndRewrite()
471 newForOp.getResults().take_front(forOp->getNumResults()); in returningMatchAndRewrite()
474 rewriter.setInsertionPointAfter(newForOp); in returningMatchAndRewrite()
483 return newForOp; in returningMatchAndRewrite()