[CIR] Fix shared library build caused by interface->IR dependence (#1900)

After #1878, we introduced a dependency from the LoopOpInterface on
BreakOp.

While here add the BreakOp handling, which will be tests by a pass
coming soon.

Co-authored-by: Tommy McMichen <[email protected]>

Author: tommymcm

clang/lib/CIR/Dialect/IR/CIRDialect.cpp

@@ -2071,6 +2071,39 @@ static void printSwitchFlatOpCases(OpAsmPrinter &p, cir::SwitchFlatOp op,
 // LoopOpInterface Methods
 //===----------------------------------------------------------------------===//
 
+void cir::LoopOpInterface::getLoopOpSuccessorRegions(
+    LoopOpInterface op, mlir::RegionBranchPoint point,
+    llvm::SmallVectorImpl<mlir::RegionSuccessor> &regions) {
+  assert(point.isParent() || point.getRegionOrNull());
+
+  // Branching to first region: go to condition or body (do-while).
+  if (point.isParent()) {
+    regions.emplace_back(&op.getEntry(), op.getEntry().getArguments());
+  }
+  // Branching from condition: go to body or exit.
+  else if (&op.getCond() == point.getRegionOrNull()) {
+    regions.emplace_back(mlir::RegionSuccessor(op->getResults()));
+    regions.emplace_back(&op.getBody(), op.getBody().getArguments());
+  }
+  // Branching from body: go to step (for) or condition.
+  else if (&op.getBody() == point.getRegionOrNull()) {
+    // If there are any breaks in the body, also go to exit.
+    op.getBody().walk([&](cir::BreakOp breakOp) {
+      if (breakOp.getBreakTarget() == op)
+        regions.emplace_back(mlir::RegionSuccessor(op->getResults()));
+    });
+
+    auto *afterBody = (op.maybeGetStep() ? op.maybeGetStep() : &op.getCond());
+    regions.emplace_back(afterBody, afterBody->getArguments());
+  }
+  // Branching from step: go to condition.
+  else if (op.maybeGetStep() == point.getRegionOrNull()) {
+    regions.emplace_back(&op.getCond(), op.getCond().getArguments());
+  } else {
+    llvm_unreachable("unexpected branch origin");
+  }
+}
+
 void cir::DoWhileOp::getSuccessorRegions(
     ::mlir::RegionBranchPoint point,
     ::llvm::SmallVectorImpl<::mlir::RegionSuccessor> &regions) {

clang/lib/CIR/Interfaces/CIRLoopOpInterface.cpp

@@ -14,34 +14,6 @@
 
 namespace cir {
 
-void LoopOpInterface::getLoopOpSuccessorRegions(
-    LoopOpInterface op, mlir::RegionBranchPoint point,
-    llvm::SmallVectorImpl<mlir::RegionSuccessor> &regions) {
-  assert(point.isParent() || point.getRegionOrNull());
-
-  // Branching to first region: go to condition or body (do-while).
-  if (point.isParent()) {
-    regions.emplace_back(&op.getEntry(), op.getEntry().getArguments());
-  }
-  // Branching from condition: go to body or exit.
-  else if (&op.getCond() == point.getRegionOrNull()) {
-    regions.emplace_back(mlir::RegionSuccessor(op->getResults()));
-    regions.emplace_back(&op.getBody(), op.getBody().getArguments());
-  }
-  // Branching from body: go to step (for) or condition.
-  else if (&op.getBody() == point.getRegionOrNull()) {
-    // FIXME(cir): Should we consider break/continue statements here?
-    auto *afterBody = (op.maybeGetStep() ? op.maybeGetStep() : &op.getCond());
-    regions.emplace_back(afterBody, afterBody->getArguments());
-  }
-  // Branching from step: go to condition.
-  else if (op.maybeGetStep() == point.getRegionOrNull()) {
-    regions.emplace_back(&op.getCond(), op.getCond().getArguments());
-  } else {
-    llvm_unreachable("unexpected branch origin");
-  }
-}
-
 /// Verify invariants of the LoopOpInterface.
 llvm::LogicalResult detail::verifyLoopOpInterface(mlir::Operation *op) {
   auto loopOp = mlir::cast<LoopOpInterface>(op);