Back to PBT run
output.generic.mlir
"builtin.module"() ({
"llvm.func"() <{CConv = #llvm.cconv<ccc>, function_type = !llvm.func<i32 (i1, i1, i32, i32)>, linkage = #llvm.linkage<external>, sym_name = "weighted_then_plain_0", unnamed_addr = 0 : i64, visibility_ = 0 : i64}> ({
^bb0(%arg4: i1, %arg5: i1, %arg6: i32, %arg7: i32):
"cf.cond_br"(%arg4)[^bb1, ^bb2] <{branch_weights = array<i32: 4, 6>, operandSegmentSizes = array<i32: 1, 0, 0>}> : (i1) -> ()
^bb1: // pred: ^bb0
"cf.br"(%arg6)[^bb3] : (i32) -> ()
^bb2: // pred: ^bb0
"cf.br"(%arg7)[^bb3] : (i32) -> ()
^bb3(%4: i32): // 2 preds: ^bb1, ^bb2
%5 = "arith.constant"() <{value = 0 : i32}> : () -> i32
%6 = "scf.if"(%arg5) ({
"scf.yield"(%4) : (i32) -> ()
}, {
"scf.yield"(%4) : (i32) -> ()
}) : (i1) -> i32
"cf.br"(%6)[^bb4] : (i32) -> ()
^bb4(%7: i32): // pred: ^bb3
"llvm.return"(%7) : (i32) -> ()
}) : () -> ()
"llvm.func"() <{CConv = #llvm.cconv<ccc>, function_type = !llvm.func<i32 (i1, i1, i32, i32)>, linkage = #llvm.linkage<external>, sym_name = "plain_then_weighted_1", unnamed_addr = 0 : i64, visibility_ = 0 : i64}> ({
^bb0(%arg0: i1, %arg1: i1, %arg2: i32, %arg3: i32):
%0 = "arith.constant"() <{value = 0 : i32}> : () -> i32
%1 = "scf.if"(%arg0) ({
"scf.yield"(%arg2) : (i32) -> ()
}, {
"scf.yield"(%arg3) : (i32) -> ()
}) : (i1) -> i32
"cf.br"(%1)[^bb1] : (i32) -> ()
^bb1(%2: i32): // pred: ^bb0
"cf.cond_br"(%arg1)[^bb2, ^bb3] <{branch_weights = array<i32: 2, 8>, operandSegmentSizes = array<i32: 1, 0, 0>}> : (i1) -> ()
^bb2: // pred: ^bb1
"cf.br"(%2)[^bb4] : (i32) -> ()
^bb3: // pred: ^bb1
"cf.br"(%2)[^bb4] : (i32) -> ()
^bb4(%3: i32): // 2 preds: ^bb2, ^bb3
"llvm.return"(%3) : (i32) -> ()
}) : () -> ()
}) : () -> ()