llvm/test/CodeGen/AMDGPU/si-annotate-nested-control-flows.ll

; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 5
; RUN: opt -mtriple=amdgpu-amd-amdhsa -p simplifycfg,amdgpu-unify-divergent-exit-nodes %s -S -o - | FileCheck %s --check-prefix=OPT
; RUN: llc -mtriple=amdgpu7.00-amd-amdhsa %s -o - | FileCheck %s --check-prefix=ISA

define void @nested_inf_loop(i1 %0, i1 %1) {
; OPT-LABEL: define void @nested_inf_loop(
; OPT-SAME: i1 [[TMP0:%.*]], i1 [[TMP1:%.*]]) {
; OPT-NEXT:  [[BB:.*:]]
; OPT-NEXT:    br label %[[BB1:.*]]
; OPT:       [[BB1]]:
; OPT-NEXT:    [[BRMERGE:%.*]] = select i1 [[TMP0]], i1 true, i1 [[TMP1]]
; OPT-NEXT:    br i1 [[BRMERGE]], label %[[BB1]], label %[[INFLOOP:.*]]
; OPT:       [[INFLOOP]]:
; OPT-NEXT:    br i1 true, label %[[INFLOOP]], label %[[DUMMYRETURNBLOCK:.*]]
; OPT:       [[DUMMYRETURNBLOCK]]:
; OPT-NEXT:    ret void
;
; ISA-LABEL: nested_inf_loop:
; ISA:       ; %bb.0: ; %BB
; ISA-NEXT:    s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0)
; ISA-NEXT:    v_and_b32_e32 v1, 1, v1
; ISA-NEXT:    v_and_b32_e32 v0, 1, v0
; ISA-NEXT:    v_cmp_eq_u32_e64 s[4:5], 1, v1
; ISA-NEXT:    v_cmp_ne_u32_e64 s[6:7], 1, v0
; ISA-NEXT:    s_mov_b64 s[8:9], 0
; ISA-NEXT:  .LBB0_1: ; %BB1
; ISA-NEXT:    ; =>This Loop Header: Depth=1
; ISA-NEXT:    ; Child Loop BB0_3 Depth 2
; ISA-NEXT:    s_and_b64 s[10:11], exec, s[6:7]
; ISA-NEXT:    s_or_b64 s[8:9], s[10:11], s[8:9]
; ISA-NEXT:    s_andn2_b64 exec, exec, s[8:9]
; ISA-NEXT:    s_cbranch_execnz .LBB0_1
; ISA-NEXT:  ; %bb.2: ; %BB2
; ISA-NEXT:    ; in Loop: Header=BB0_1 Depth=1
; ISA-NEXT:    s_or_b64 exec, exec, s[8:9]
; ISA-NEXT:    s_mov_b64 s[8:9], 0
; ISA-NEXT:  .LBB0_3: ; %BB4
; ISA-NEXT:    ; Parent Loop BB0_1 Depth=1
; ISA-NEXT:    ; => This Inner Loop Header: Depth=2
; ISA-NEXT:    s_and_b64 s[10:11], exec, s[4:5]
; ISA-NEXT:    s_or_b64 s[8:9], s[10:11], s[8:9]
; ISA-NEXT:    s_andn2_b64 exec, exec, s[8:9]
; ISA-NEXT:    s_cbranch_execnz .LBB0_3
; ISA-NEXT:  ; %bb.4: ; %loop.exit.guard
; ISA-NEXT:    ; in Loop: Header=BB0_1 Depth=1
; ISA-NEXT:    s_or_b64 exec, exec, s[8:9]
; ISA-NEXT:    s_mov_b64 s[8:9], 0
; ISA-NEXT:    s_branch .LBB0_1
; ISA-NEXT:  ; %bb.5: ; %DummyReturnBlock
; ISA-NEXT:    s_setpc_b64 s[30:31]
BB:
  br label %BB1

BB1:
  br i1 %0, label %BB3, label %BB2

BB2:
  br label %BB4

BB4:
  br i1 %1, label %BB3, label %BB4

BB3:
  br label %BB1
}

define void @nested_inf_loop_callbr(i32 %0, i32 %1) {
; OPT-LABEL: define void @nested_inf_loop_callbr(
; OPT-SAME: i32 [[TMP0:%.*]], i32 [[TMP1:%.*]]) {
; OPT-NEXT:  [[BB:.*:]]
; OPT-NEXT:    callbr void asm "", ""()
; OPT-NEXT:            to label %[[BB1:.*]] []
; OPT:       [[BB1]]:
; OPT-NEXT:    callbr void asm "", "r,!i"(i32 [[TMP0]])
; OPT-NEXT:            to label %[[BB3:.*]] [label %BB2]
; OPT:       [[BB2:.*:]]
; OPT-NEXT:    callbr void asm "", ""()
; OPT-NEXT:            to label %[[BB4:.*]] []
; OPT:       [[BB4]]:
; OPT-NEXT:    br i1 true, label %[[TRANSITIONBLOCK:.*]], label %[[DUMMYRETURNBLOCK:.*]]
; OPT:       [[TRANSITIONBLOCK]]:
; OPT-NEXT:    callbr void asm "", "r,!i"(i32 [[TMP1]])
; OPT-NEXT:            to label %[[BB3]] [label %BB4]
; OPT:       [[BB3]]:
; OPT-NEXT:    callbr void asm "", ""()
; OPT-NEXT:            to label %[[BB1]] []
; OPT:       [[DUMMYRETURNBLOCK]]:
; OPT-NEXT:    ret void
;
; ISA-LABEL: nested_inf_loop_callbr:
; ISA:       ; %bb.0: ; %BB
; ISA-NEXT:    s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0)
; ISA-NEXT:    ;;#ASMSTART
; ISA-NEXT:    ;;#ASMEND
; ISA-NEXT:    s_and_b64 s[4:5], exec, 0
; ISA-NEXT:  .LBB1_1: ; %BB1
; ISA-NEXT:    ; =>This Loop Header: Depth=1
; ISA-NEXT:    ; Child Loop BB1_4 Depth 2
; ISA-NEXT:    ;;#ASMSTART
; ISA-NEXT:    ;;#ASMEND
; ISA-NEXT:  .LBB1_2: ; %BB3
; ISA-NEXT:    ; in Loop: Header=BB1_1 Depth=1
; ISA-NEXT:    ;;#ASMSTART
; ISA-NEXT:    ;;#ASMEND
; ISA-NEXT:    s_branch .LBB1_1
; ISA-NEXT:  .LBB1_3: ; Inline asm indirect target
; ISA-NEXT:    ; %BB2
; ISA-NEXT:    ; in Loop: Header=BB1_1 Depth=1
; ISA-NEXT:    ; Label of block must be emitted
; ISA-NEXT:    ;;#ASMSTART
; ISA-NEXT:    ;;#ASMEND
; ISA-NEXT:  .LBB1_4: ; Inline asm indirect target
; ISA-NEXT:    ; %BB4
; ISA-NEXT:    ; Parent Loop BB1_1 Depth=1
; ISA-NEXT:    ; => This Inner Loop Header: Depth=2
; ISA-NEXT:    ; Label of block must be emitted
; ISA-NEXT:    s_mov_b64 vcc, s[4:5]
; ISA-NEXT:    s_cbranch_vccnz .LBB1_7
; ISA-NEXT:  ; %bb.5: ; %TransitionBlock
; ISA-NEXT:    ; in Loop: Header=BB1_4 Depth=2
; ISA-NEXT:    ;;#ASMSTART
; ISA-NEXT:    ;;#ASMEND
; ISA-NEXT:  ; %bb.6: ; %TransitionBlock.target.BB3
; ISA-NEXT:    ; in Loop: Header=BB1_1 Depth=1
; ISA-NEXT:    s_branch .LBB1_2
; ISA-NEXT:  .LBB1_7: ; in Loop: Header=BB1_1 Depth=1
; ISA-NEXT:    s_cbranch_execz .LBB1_2
; ISA-NEXT:  ; %bb.8: ; %DummyReturnBlock
; ISA-NEXT:    s_setpc_b64 s[30:31]
BB:
  callbr void asm "", ""() to label %BB1 []

BB1:
  callbr void asm "", "r,!i"(i32 %0) to label %BB3 [label %BB2]

BB2:
  callbr void asm "", ""() to label %BB4 []

BB4:
  callbr void asm "", "r,!i"(i32 %1) to label %BB3 [label %BB4]

BB3:
  callbr void asm "", ""() to label %BB1 []
}