Back to PBT run

input.mlir

#loop_ann = #llvm.loop_annotation<mustProgress = true>

llvm.func @plain_then_weighted_0(%plain: i1, %weighted: i1, %a: i32, %b: i32) -> i32 {
  cf.cond_br %plain, ^plain_true, ^plain_false
^plain_true:
  cf.br ^weighted_entry(%a : i32)
^plain_false:
  cf.br ^weighted_entry(%b : i32)
^weighted_entry(%seed: i32):
  cf.cond_br %weighted weights([9, 1]), ^weighted_true, ^weighted_false
^weighted_true:
  cf.br ^exit(%seed : i32)
^weighted_false:
  cf.br ^exit(%seed : i32)
^exit(%r: i32):
  llvm.return %r : i32
}

llvm.func @plain_then_weighted_1(%plain: i1, %weighted: i1, %a: i32, %b: i32) -> i32 {
  cf.cond_br %plain, ^plain_true, ^plain_false
^plain_true:
  cf.br ^weighted_entry(%a : i32)
^plain_false:
  cf.br ^weighted_entry(%b : i32)
^weighted_entry(%seed: i32):
  cf.cond_br %weighted weights([7, 3]), ^weighted_true, ^weighted_false
^weighted_true:
  cf.br ^exit(%seed : i32)
^weighted_false:
  cf.br ^exit(%seed : i32)
^exit(%r: i32):
  llvm.return %r : i32
}