[CIR][NFC] Add getBreakTarget method to cir::BreakOp

Author: tommymcm

clang/include/clang/CIR/Dialect/IR/CIROps.td

@@ -1053,6 +1053,10 @@ def CIR_BreakOp : CIR_Op<"break", [Terminator]> {
   }];
   let assemblyFormat = "attr-dict";
   let hasVerifier = 1;
+  let extraClassDeclaration = [{
+    /// Find the parent operation corresponding to this break.
+    mlir::Operation *getBreakTarget();
+    }];
 }
 
 //===----------------------------------------------------------------------===//

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

@@ -324,6 +324,15 @@ void cir::AllocaOp::build(::mlir::OpBuilder &odsBuilder,
 // BreakOp
 //===----------------------------------------------------------------------===//
 
+mlir::Operation *cir::BreakOp::getBreakTarget() {
+  // Find the innermost loop or switch parent operation.
+  mlir::Operation *parentOp = getOperation()->getParentOp();
+  while (::mlir::isa_and_present<cir::LoopOpInterface, cir::SwitchOp>(parentOp))
+    parentOp = parentOp->getParentOp();
+
+  return parentOp;
+}
+
 LogicalResult cir::BreakOp::verify() {
   if (!getOperation()->getParentOfType<LoopOpInterface>() &&
       !getOperation()->getParentOfType<SwitchOp>())