Back to PBT run

input.mlir

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

llvm.func @weighted_then_cycle_0(%weighted: i1, %limit: i32) -> i32 {
  %zero = arith.constant 0 : i32
  %one = arith.constant 1 : i32
  cf.cond_br %weighted weights([4, 6]), ^left, ^right
^left:
  cf.br ^header(%zero : i32)
^right:
  cf.br ^header(%one : i32)
^header(%iv: i32):
  %done = arith.cmpi eq, %iv, %limit : i32
  cf.cond_br %done, ^exit, ^latch
^latch:
  %next = arith.addi %iv, %one : i32
  cf.br ^header(%next : i32) {llvm.loop_annotation = #loop_ann}
^exit:
  llvm.return %iv : i32
}

llvm.func @nested_weighted_arm_1(%weighted: i1, %plain: i1, %a: i32, %b: i32) -> i32 {
  cf.cond_br %weighted weights([1, 9]), ^left, ^right
^left:
  cf.cond_br %plain, ^left_true, ^left_false
^left_true:
  cf.br ^exit(%a : i32)
^left_false:
  cf.br ^exit(%b : i32)
^right:
  cf.br ^exit(%b : i32)
^exit(%r: i32):
  llvm.return %r : i32
}

llvm.func @plain_diamond_2(%c: i1, %a: i32, %b: i32) -> i32 {
  cf.cond_br %c, ^yes, ^no
^yes:
  cf.br ^exit(%a : i32)
^no:
  cf.br ^exit(%b : i32)
^exit(%r: i32):
  llvm.return %r : i32
}

llvm.func @counted_cycle_3(%limit: i32) -> i32 {
  %zero = arith.constant 0 : i32
  %step = arith.constant 2 : i32
  cf.br ^header(%zero : i32)
^header(%iv: i32):
  %done = arith.cmpi eq, %iv, %limit : i32
  cf.cond_br %done, ^exit, ^latch
^latch:
  %next = arith.addi %iv, %step : i32
  cf.br ^header(%next : i32) {llvm.loop_annotation = #loop_ann}
^exit:
  llvm.return %iv : i32
}