Lines Matching defs:forallOp
274 scf::ParallelOp forallOp,
310 Value upper = irMap.lookup(forallOp.getUpperBound()[0]);
316 rewriter.cloneRegionBefore(forallOp.getRegion(), forOp.getRegion(),
1164 LogicalResult matchAndRewrite(scf::ParallelOp forallOp,
1171 if (!forallOp->hasAttr(LoopEmitter::getLoopEmitterLoopAttrName()) ||
1172 forallOp.getNumReductions() != 0 || forallOp.getNumLoops() != 1 ||
1173 !matchPattern(forallOp.getLowerBound()[0], m_Zero()) ||
1174 !matchPattern(forallOp.getStep()[0], m_One()))
1178 forallOp->walk([&](Operation *op) {
1187 if (!forallOp.getRegion().findAncestorBlockInRegion(*block))
1211 Location loc = forallOp->getLoc();
1218 ModuleOp topModule = forallOp->getParentOfType<ModuleOp>();
1221 genGPUCode(rewriter, gpuFunc, forallOp, constants, scalars, buffers);
1235 rewriter.eraseOp(forallOp);