Lines Matching defs:topLevelForallOp

637                                            scf::ForallOp &topLevelForallOp,
642     if (topLevelForallOp)
645     topLevelForallOp = forallOp;
649   if (walkResult.wasInterrupted() || !topLevelForallOp)
670   scf::ForallOp topLevelForallOp;
672       target, topLevelForallOp, transformOp);
677   assert(topLevelForallOp && "expect an scf.forall");
684   rewriter.setInsertionPoint(topLevelForallOp);
694     Operation *newForallOp = rewriter.clone(*topLevelForallOp);
695     rewriter.eraseOp(topLevelForallOp);
696     topLevelForallOp = cast<scf::ForallOp>(newForallOp);
701   if (topLevelForallOp.getMapping()) {
703         topLevelForallOp.getMapping()->getValue().front());
709       rewriter, transformOp, topLevelForallOp, gridDims, gpuBlockIdBuilder);