Lines Matching defs:loopInfo

730   const LoopInfo &loopInfo = loopStack.back();
732     auto iterateOp = llvm::cast<IterateOp>(loopInfo.loop);
741   if (auto forOp = llvm::dyn_cast<scf::ForOp>(loopInfo.loop)) {
751     auto parOp = llvm::cast<scf::ParallelOp>(loopInfo.loop);
806   const LoopInfo &loopInfo = loopStack.back();
807   auto whileOp = llvm::cast<scf::WhileOp>(loopInfo.loop);
808   Value iv = loopInfo.iv;
819   for (auto [tid, lvl] : unpackTensorLevelRange(loopInfo.tidLvls)) {
865   const LoopInfo &loopInfo = loopStack.back();
867     Operation *p = loopInfo.loop;
880   rewriter.setInsertionPointToEnd(loopInfo.userCodeBlock);
881   if (!loopInfo.userCodeBlock->empty() &&
882       llvm::isa<scf::YieldOp>(&loopInfo.userCodeBlock->back())) {
885     assert(loopInfo.userCodeBlock->back().getNumResults() == 0);
886     rewriter.setInsertionPoint(&loopInfo.userCodeBlock->back());
889   if (llvm::isa<scf::WhileOp>(loopInfo.loop)) {