From cf14ef983a49e47ac49e1a9f64cf0849c8b664bc Mon Sep 17 00:00:00 2001
From: Jeff Niu <jeff@modular.com>
Date: Wed, 11 Jan 2023 10:03:10 -0800
Subject: [PATCH] [mlir][llvm] Add a convenience builder to BrOp

This builder takes no successor arguments.

Reviewed By: mehdi_amini

Differential Revision: https://reviews.llvm.org/D141522
---
 mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td b/mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td
index bfb2226..f426d69 100644
--- a/mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td
+++ b/mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td
@@ -772,7 +772,12 @@ def LLVM_BrOp : LLVM_TerminatorOp<"br",
   let assemblyFormat = [{
     $dest (`(` $destOperands^ `:` type($destOperands) `)`)? attr-dict
   }];
-  let builders = [LLVM_TerminatorPassthroughOpBuilder];
+  let builders = [
+    OpBuilder<(ins "Block *":$dest), [{
+      build($_builder, $_state, ValueRange(), dest);
+    }]>,
+    LLVM_TerminatorPassthroughOpBuilder
+  ];
 }
 def LLVM_CondBrOp : LLVM_TerminatorOp<"cond_br",
     [AttrSizedOperandSegments, DeclareOpInterfaceMethods<BranchOpInterface>,
-- 
2.7.4