// RUN: iree-compile \
// RUN:   %s \
// RUN:   --compile-mode=vm \
// RUN:   -o=%t.vmfb && \
// RUN: iree-dump-module \
// RUN:   --output=disassembly \
// RUN:   %t.vmfb | \
// RUN: FileCheck --check-prefix=CHECK %s
// RUN: iree-dump-module \
// RUN:   --output=disassembly \
// RUN:   --bytecode_offsets \
// RUN:   %t.vmfb | \
// RUN: FileCheck --check-prefix=OFFSETS %s

vm.module @dump_control_flow {
  vm.export @branch
  vm.func @branch(%cond: i32) -> i32 {
    %c1 = vm.const.i32 1
    %c2 = vm.const.i32 2
    vm.cond_br %cond, ^left, ^right
  ^left:
    vm.br ^exit(%c1 : i32)
  ^right:
    vm.br ^exit(%c2 : i32)
  ^exit(%value : i32):
    vm.return %value : i32
  }
}

// CHECK: vm.module @dump_control_flow version 0
// CHECK-NEXT: vm.type i32
// CHECK-EMPTY:
// CHECK-NEXT: vm.export @branch
// CHECK-NEXT: vm.func @branch(%i0: i32) -> (i32) {
// CHECK: ^bb0:
// CHECK:   vm.cond_br %i0, ^bb1({{.*}}), ^bb1({{.*}})
// CHECK: ^bb1:
// OFFSETS: [00000000]^bb0: