llvm/llvm/test/Transforms/CodeGenPrepare/AArch64/large-offset-gep.ll

; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
; RUN: llc -mtriple=aarch64-linux-gnu -verify-machineinstrs -o - %s | FileCheck %s

%struct_type = type { [10000 x i32], i32, i32 }

define void @test1(ptr %s, i32 %n) {
; CHECK-LABEL: test1:
; CHECK:       // %bb.0: // %entry
; CHECK-NEXT:    ldr x8, [x0]
; CHECK-NEXT:    mov w9, wzr
; CHECK-NEXT:    add x8, x8, #9, lsl #12 // =36864
; CHECK-NEXT:    cmp w9, w1
; CHECK-NEXT:    b.ge .LBB0_2
; CHECK-NEXT:  .LBB0_1: // %while_body
; CHECK-NEXT:    // =>This Inner Loop Header: Depth=1
; CHECK-NEXT:    str w9, [x8, #3140]
; CHECK-NEXT:    add w9, w9, #1
; CHECK-NEXT:    str w9, [x8, #3136]
; CHECK-NEXT:    cmp w9, w1
; CHECK-NEXT:    b.lt .LBB0_1
; CHECK-NEXT:  .LBB0_2: // %while_end
; CHECK-NEXT:    ret
entry:
  %struct = load ptr, ptr %s
  br label %while_cond

while_cond:
  %phi = phi i32 [ 0, %entry ], [ %i, %while_body ]
  %gep0 = getelementptr %struct_type, ptr %struct, i64 0, i32 1
  %gep1 = getelementptr %struct_type, ptr %struct, i64 0, i32 2
  %cmp = icmp slt i32 %phi, %n
  br i1 %cmp, label %while_body, label %while_end

while_body:
  %i = add i32 %phi, 1
  store i32 %i, ptr %gep0
  store i32 %phi, ptr %gep1
  br label %while_cond

while_end:
  ret void
}

define void @test2(ptr %struct, i32 %n) {
; CHECK-LABEL: test2:
; CHECK:       // %bb.0: // %entry
; CHECK-NEXT:    cbz x0, .LBB1_3
; CHECK-NEXT:  // %bb.1: // %while_cond.preheader
; CHECK-NEXT:    mov w9, wzr
; CHECK-NEXT:    add x8, x0, #9, lsl #12 // =36864
; CHECK-NEXT:    cmp w9, w1
; CHECK-NEXT:    b.ge .LBB1_3
; CHECK-NEXT:  .LBB1_2: // %while_body
; CHECK-NEXT:    // =>This Inner Loop Header: Depth=1
; CHECK-NEXT:    str w9, [x8, #3140]
; CHECK-NEXT:    add w9, w9, #1
; CHECK-NEXT:    str w9, [x8, #3136]
; CHECK-NEXT:    cmp w9, w1
; CHECK-NEXT:    b.lt .LBB1_2
; CHECK-NEXT:  .LBB1_3: // %while_end
; CHECK-NEXT:    ret
entry:
  %cmp = icmp eq ptr %struct, null
  br i1 %cmp, label %while_end, label %while_cond

while_cond:
  %phi = phi i32 [ 0, %entry ], [ %i, %while_body ]
  %gep0 = getelementptr %struct_type, ptr %struct, i64 0, i32 1
  %gep1 = getelementptr %struct_type, ptr %struct, i64 0, i32 2
  %cmp1 = icmp slt i32 %phi, %n
  br i1 %cmp1, label %while_body, label %while_end

while_body:
  %i = add i32 %phi, 1
  store i32 %i, ptr %gep0
  store i32 %phi, ptr %gep1
  br label %while_cond

while_end:
  ret void
}

define void @test3(ptr %s1, ptr %s2, i1 %cond, i32 %n) {
; CHECK-LABEL: test3:
; CHECK:       // %bb.0: // %entry
; CHECK-NEXT:    tst w2, #0x1
; CHECK-NEXT:    csel x8, x1, x0, ne
; CHECK-NEXT:    cbz x8, .LBB2_3
; CHECK-NEXT:  // %bb.1: // %while_cond.preheader
; CHECK-NEXT:    mov w9, wzr
; CHECK-NEXT:    add x8, x8, #9, lsl #12 // =36864
; CHECK-NEXT:    cmp w9, w3
; CHECK-NEXT:    b.ge .LBB2_3
; CHECK-NEXT:  .LBB2_2: // %while_body
; CHECK-NEXT:    // =>This Inner Loop Header: Depth=1
; CHECK-NEXT:    str w9, [x8, #3140]
; CHECK-NEXT:    add w9, w9, #1
; CHECK-NEXT:    str w9, [x8, #3136]
; CHECK-NEXT:    cmp w9, w3
; CHECK-NEXT:    b.lt .LBB2_2
; CHECK-NEXT:  .LBB2_3: // %while_end
; CHECK-NEXT:    ret
entry:
  br i1 %cond, label %if_true, label %if_end

if_true:
  br label %if_end

if_end:
  %struct = phi ptr [ %s1, %entry ], [ %s2, %if_true ]
  %cmp = icmp eq ptr %struct, null
  br i1 %cmp, label %while_end, label %while_cond

while_cond:
  %phi = phi i32 [ 0, %if_end ], [ %i, %while_body ]
  %gep0 = getelementptr %struct_type, ptr %struct, i64 0, i32 1
  %gep1 = getelementptr %struct_type, ptr %struct, i64 0, i32 2
  %cmp1 = icmp slt i32 %phi, %n
  br i1 %cmp1, label %while_body, label %while_end

while_body:
  %i = add i32 %phi, 1
  store i32 %i, ptr %gep0
  store i32 %phi, ptr %gep1
  br label %while_cond

while_end:
  ret void
}

declare ptr @foo()
declare void @foo2()

define void @test4(i32 %n) uwtable personality ptr @__FrameHandler {
; CHECK-LABEL: test4:
; CHECK:       .Lfunc_begin0:
; CHECK-NEXT:    .cfi_startproc
; CHECK-NEXT:    .cfi_personality 156, DW.ref.__FrameHandler
; CHECK-NEXT:    .cfi_lsda 28, .Lexception0
; CHECK-NEXT:  // %bb.0: // %entry
; CHECK-NEXT:    str x30, [sp, #-32]! // 8-byte Folded Spill
; CHECK-NEXT:    .cfi_def_cfa_offset 32
; CHECK-NEXT:    stp x20, x19, [sp, #16] // 16-byte Folded Spill
; CHECK-NEXT:    .cfi_offset w19, -8
; CHECK-NEXT:    .cfi_offset w20, -16
; CHECK-NEXT:    .cfi_offset w30, -32
; CHECK-NEXT:    .cfi_remember_state
; CHECK-NEXT:    mov w19, w0
; CHECK-NEXT:    mov w20, wzr
; CHECK-NEXT:  .LBB3_1: // %while_cond
; CHECK-NEXT:    // =>This Inner Loop Header: Depth=1
; CHECK-NEXT:  .Ltmp0:
; CHECK-NEXT:    bl foo
; CHECK-NEXT:  .Ltmp1:
; CHECK-NEXT:  // %bb.2: // %while_cond_x.split
; CHECK-NEXT:    // in Loop: Header=BB3_1 Depth=1
; CHECK-NEXT:    add x8, x0, #9, lsl #12 // =36864
; CHECK-NEXT:    cmp w20, w19
; CHECK-NEXT:    str wzr, [x8, #3136]
; CHECK-NEXT:    b.ge .LBB3_4
; CHECK-NEXT:  // %bb.3: // %while_body
; CHECK-NEXT:    // in Loop: Header=BB3_1 Depth=1
; CHECK-NEXT:    str w20, [x8, #3140]
; CHECK-NEXT:    add w20, w20, #1
; CHECK-NEXT:    str w20, [x8, #3136]
; CHECK-NEXT:    b .LBB3_1
; CHECK-NEXT:  .LBB3_4: // %while_end
; CHECK-NEXT:    ldp x20, x19, [sp, #16] // 16-byte Folded Reload
; CHECK-NEXT:    ldr x30, [sp], #32 // 8-byte Folded Reload
; CHECK-NEXT:    .cfi_def_cfa_offset 0
; CHECK-NEXT:    .cfi_restore w19
; CHECK-NEXT:    .cfi_restore w20
; CHECK-NEXT:    .cfi_restore w30
; CHECK-NEXT:    ret
; CHECK-NEXT:  .LBB3_5: // %cleanup
; CHECK-NEXT:    .cfi_restore_state
; CHECK-NEXT:  .Ltmp2:
; CHECK-NEXT:    mov x19, x0
; CHECK-NEXT:    bl foo2
; CHECK-NEXT:    mov x0, x19
; CHECK-NEXT:    bl _Unwind_Resume
entry:
  br label %while_cond

while_cond:
  %phi = phi i32 [ 0, %entry ], [ %i, %while_body ]
  %struct = invoke ptr @foo() to label %while_cond_x unwind label %cleanup

while_cond_x:
  %gep0 = getelementptr %struct_type, ptr %struct, i64 0, i32 1
  %gep1 = getelementptr %struct_type, ptr %struct, i64 0, i32 2
  store i32 0, ptr %gep0
  %cmp = icmp slt i32 %phi, %n
  br i1 %cmp, label %while_body, label %while_end

while_body:
  %i = add i32 %phi, 1
  store i32 %i, ptr %gep0
  store i32 %phi, ptr %gep1
  br label %while_cond

while_end:
  ret void

cleanup:
  %x10 = landingpad { ptr, i32 }
          cleanup
  call void @foo2()
  resume { ptr, i32 } %x10
}

declare i32 @__FrameHandler(...)

define void @test5(ptr %s, i32 %n) {
; CHECK-LABEL: test5:
; CHECK:       // %bb.0: // %entry
; CHECK-NEXT:    ldr x8, [x0]
; CHECK-NEXT:    mov w9, wzr
; CHECK-NEXT:    add x8, x8, #19, lsl #12 // =77824
; CHECK-NEXT:    cmp w9, w1
; CHECK-NEXT:    b.ge .LBB4_2
; CHECK-NEXT:  .LBB4_1: // %while_body
; CHECK-NEXT:    // =>This Inner Loop Header: Depth=1
; CHECK-NEXT:    str w9, [x8, #2180]
; CHECK-NEXT:    add w9, w9, #1
; CHECK-NEXT:    str w9, [x8, #2176]
; CHECK-NEXT:    cmp w9, w1
; CHECK-NEXT:    b.lt .LBB4_1
; CHECK-NEXT:  .LBB4_2: // %while_end
; CHECK-NEXT:    ret
entry:
  %struct = load ptr, ptr %s
  br label %while_cond

while_cond:
  %phi = phi i32 [ 0, %entry ], [ %i, %while_body ]
  %gep0 = getelementptr [65536 x i32], ptr %struct, i64 0, i32 20000
  %gep1 = getelementptr [65536 x i32], ptr %struct, i64 0, i32 20001
  %cmp = icmp slt i32 %phi, %n
  br i1 %cmp, label %while_body, label %while_end

while_body:
  %i = add i32 %phi, 1
  store i32 %i, ptr %gep0
  store i32 %phi, ptr %gep1
  br label %while_cond

while_end:
  ret void
}

declare ptr @llvm.strip.invariant.group.p0(ptr)

define void @test_invariant_group(i32 %arg, i1 %c) {
; CHECK-LABEL: test_invariant_group:
; CHECK:       // %bb.0: // %bb
; CHECK-NEXT:    tbz w1, #0, .LBB5_4
; CHECK-NEXT:  // %bb.1: // %bb6
; CHECK-NEXT:    cbz w0, .LBB5_3
; CHECK-NEXT:  .LBB5_2: // %bb1
; CHECK-NEXT:    // =>This Inner Loop Header: Depth=1
; CHECK-NEXT:    tbnz w1, #0, .LBB5_2
; CHECK-NEXT:    b .LBB5_4
; CHECK-NEXT:  .LBB5_3: // %bb2
; CHECK-NEXT:    tbz w1, #0, .LBB5_5
; CHECK-NEXT:  .LBB5_4: // %bb5
; CHECK-NEXT:    ret
; CHECK-NEXT:  .LBB5_5: // %bb4
; CHECK-NEXT:    mov w8, #1 // =0x1
; CHECK-NEXT:    str x8, [x8]
; CHECK-NEXT:    ret
bb:
  br i1 %c, label %bb6, label %bb5

bb1:                                              ; preds = %bb6, %bb1
  br i1 %c, label %bb1, label %bb5

bb2:                                              ; preds = %bb6
  %i = getelementptr inbounds i8, ptr %i7, i32 40000
  br i1 %c, label %bb5, label %bb4

bb4:                                              ; preds = %bb2
  store i64 1, ptr %i, align 8
  br label %bb5

bb5:                                              ; preds = %bb4, %bb2, %bb1, %bb
  ret void

bb6:                                              ; preds = %bb
  %i7 = call ptr @llvm.strip.invariant.group.p0(ptr nonnull undef)
  %i8 = icmp eq i32 %arg, 0
  br i1 %i8, label %bb2, label %bb1
}