From ad0a96d09abe14f207d2ca3bc0966cb46bac2c7d Mon Sep 17 00:00:00 2001
From: Thomas Alexander <talexander@nvidia.com>
Date: Wed, 14 Jan 2026 11:49:11 -0400
Subject: [PATCH] Fix use-after-free in LoopUnrollPatterns.inc

Signed-off-by: Thomas Alexander <talexander@nvidia.com>
---
 lib/Optimizer/Transforms/LoopUnrollPatterns.inc | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/lib/Optimizer/Transforms/LoopUnrollPatterns.inc b/lib/Optimizer/Transforms/LoopUnrollPatterns.inc
index 7899cf644c1..210ff9e3eb1 100644
--- a/lib/Optimizer/Transforms/LoopUnrollPatterns.inc
+++ b/lib/Optimizer/Transforms/LoopUnrollPatterns.inc
@@ -147,8 +147,7 @@ struct UnrollCountedLoop : public OpRewritePattern<cudaq::cc::LoopOp> {
           auto termOpers = cont.getOperands();
           rewriter.setInsertionPoint(cont);
           rewriter.replaceOpWithNewOp<cf::BranchOp>(cont, contBlock, termOpers);
-        }
-        if (allowBreak) {
+        } else if (allowBreak) {
           if (auto brk = dyn_cast<cudaq::cc::BreakOp>(term)) {
             auto termOpers = brk.getOperands();
             rewriter.setInsertionPoint(brk);