Method Process

source/opt/loop_unroller.cpp:1180–1215

 */

Pass::Status LoopUnroller::Process() {
  bool changed = false;
  for (Function& f : *context()->module()) {
    if (f.IsDeclaration()) {
      continue;
    }

    LoopDescriptor* LD = context()->GetLoopDescriptor(&f);
    for (Loop& loop : *LD) {
      LoopUtils loop_utils{context(), &loop};
      if (!loop.HasUnrollLoopControl() || !loop_utils.CanPerformUnroll()) {
        continue;
      }

      if (fully_unroll_) {
        if (!loop_utils.FullyUnroll()) {
          return Status::Failure;
        }
        changed = true;
      } else {
        if (!loop_utils.PartiallyUnroll(unroll_factor_)) {
          return Status::Failure;
        }
        changed = true;
      }
    }
    LD->PostModificationCleanup();
  }

  if (changed) {
    context()->InvalidateAnalysesExceptFor(
        IRContext::Analysis::kAnalysisLoopAnalysis);
  }

  return changed ? Status::SuccessWithChange : Status::SuccessWithoutChange;
}

}  // namespace opt
}  // namespace spvtools