test/CodeGen/AArch64/arm64-variadic-aapcs.ll

   1 ; RUN: llc -verify-machineinstrs -mtriple=arm64-linux-gnu -pre-RA-sched=linearize -enable-misched=false -disable-post-ra < %s | FileCheck %s
   2
   3 %va_list = type {i8*, i8*, i8*, i32, i32}
   4
   5 @var = global %va_list zeroinitializer, align 8
   6
   7 declare void @llvm.va_start(i8*)
   8
   9 define void @test_simple(i32 %n, ...) {
  10 ; CHECK-LABEL: test_simple:
  11 ; CHECK: sub sp, sp, #[[STACKSIZE:[0-9]+]]
  12 ; CHECK: add [[STACK_TOP:x[0-9]+]], sp, #[[STACKSIZE]]
  13
  14 ; CHECK: adrp x[[VA_LIST_HI:[0-9]+]], var
  15 ; CHECK: add x[[VA_LIST:[0-9]+]], {{x[0-9]+}}, :lo12:var
  16
  17 ; CHECK-DAG: stp x6, x7, [sp, #
  18 ; ... omit middle ones ...
  19 ; CHECK-DAG: str x1, [sp, #[[GR_BASE:[0-9]+]]]
  20
  21 ; CHECK-DAG: stp q0, q1, [sp]
  22 ; ... omit middle ones ...
  23 ; CHECK-DAG: stp q6, q7, [sp, #
  24
  25 ; CHECK: str [[STACK_TOP]], [x[[VA_LIST]]]
  26
  27 ; CHECK: add [[GR_TOPTMP:x[0-9]+]], sp, #[[GR_BASE]]
  28 ; CHECK: add [[GR_TOP:x[0-9]+]], [[GR_TOPTMP]], #56
  29 ; CHECK: str [[GR_TOP]], [x[[VA_LIST]], #8]
  30
  31 ; CHECK: mov [[VR_TOPTMP:x[0-9]+]], sp
  32 ; CHECK: add [[VR_TOP:x[0-9]+]], [[VR_TOPTMP]], #128
  33 ; CHECK: str [[VR_TOP]], [x[[VA_LIST]], #16]
  34
  35 ; CHECK: mov     [[GRVR:x[0-9]+]], #-56
  36 ; CHECK: movk    [[GRVR]], #65408, lsl #32
  37 ; CHECK: str     [[GRVR]], [x[[VA_LIST]], #24]
  38
  39   %addr = bitcast %va_list* @var to i8*
  40   call void @llvm.va_start(i8* %addr)
  41
  42   ret void
  43 }
  44
  45 define void @test_fewargs(i32 %n, i32 %n1, i32 %n2, float %m, ...) {
  46 ; CHECK-LABEL: test_fewargs:
  47 ; CHECK: sub sp, sp, #[[STACKSIZE:[0-9]+]]
  48 ; CHECK: add [[STACK_TOP:x[0-9]+]], sp, #[[STACKSIZE]]
  49
  50 ; CHECK: adrp x[[VA_LIST_HI:[0-9]+]], var
  51 ; CHECK: add x[[VA_LIST:[0-9]+]], {{x[0-9]+}}, :lo12:var
  52
  53 ; CHECK-DAG: stp x6, x7, [sp, #
  54 ; ... omit middle ones ...
  55 ; CHECK-DAG: str x3, [sp, #[[GR_BASE:[0-9]+]]]
  56
  57 ; CHECK-DAG: stp q6, q7, [sp, #80]
  58 ; ... omit middle ones ...
  59 ; CHECK-DAG: str q1, [sp]
  60
  61 ; CHECK: str [[STACK_TOP]], [x[[VA_LIST]]]
  62
  63 ; CHECK: add [[GR_TOPTMP:x[0-9]+]], sp, #[[GR_BASE]]
  64 ; CHECK: add [[GR_TOP:x[0-9]+]], [[GR_TOPTMP]], #40
  65 ; CHECK: str [[GR_TOP]], [x[[VA_LIST]], #8]
  66
  67 ; CHECK: mov [[VR_TOPTMP:x[0-9]+]], sp
  68 ; CHECK: add [[VR_TOP:x[0-9]+]], [[VR_TOPTMP]], #112
  69 ; CHECK: str [[VR_TOP]], [x[[VA_LIST]], #16]
  70
  71 ; CHECK: mov  [[GRVR_OFFS:x[0-9]+]], #-40
  72 ; CHECK: movk [[GRVR_OFFS]], #65424, lsl #32
  73 ; CHECK: str  [[GRVR_OFFS]], [x[[VA_LIST]], #24]
  74
  75   %addr = bitcast %va_list* @var to i8*
  76   call void @llvm.va_start(i8* %addr)
  77
  78   ret void
  79 }
  80
  81 define void @test_nospare([8 x i64], [8 x float], ...) {
  82 ; CHECK-LABEL: test_nospare:
  83
  84   %addr = bitcast %va_list* @var to i8*
  85   call void @llvm.va_start(i8* %addr)
  86 ; CHECK-NOT: sub sp, sp
  87 ; CHECK: mov [[STACK:x[0-9]+]], sp
  88 ; CHECK: add x[[VAR:[0-9]+]], {{x[0-9]+}}, :lo12:var
  89 ; CHECK: str [[STACK]], [x[[VAR]]]
  90
  91   ret void
  92 }
  93
  94 ; If there are non-variadic arguments on the stack (here two i64s) then the
  95 ; __stack field should point just past them.
  96 define void @test_offsetstack([8 x i64], [2 x i64], [3 x float], ...) {
  97 ; CHECK-LABEL: test_offsetstack:
  98
  99 ; CHECK-DAG: stp {{q[0-9]+}}, {{q[0-9]+}}, [sp, #48]
 100 ; CHECK-DAG: stp {{q[0-9]+}}, {{q[0-9]+}}, [sp, #16]
 101 ; CHECK-DAG: str {{q[0-9]+}}, [sp]
 102 ; CHECK-DAG: add [[STACK_TOP:x[0-9]+]], sp, #96
 103 ; CHECK-DAG: add x[[VAR:[0-9]+]], {{x[0-9]+}}, :lo12:var
 104 ; CHECK-DAG: str [[STACK_TOP]], [x[[VAR]]]
 105
 106   %addr = bitcast %va_list* @var to i8*
 107   call void @llvm.va_start(i8* %addr)
 108   ret void
 109 }
 110
 111 declare void @llvm.va_end(i8*)
 112
 113 define void @test_va_end() nounwind {
 114 ; CHECK-LABEL: test_va_end:
 115 ; CHECK-NEXT: %bb.0
 116
 117   %addr = bitcast %va_list* @var to i8*
 118   call void @llvm.va_end(i8* %addr)
 119
 120   ret void
 121 ; CHECK-NEXT: ret
 122 }
 123
 124 declare void @llvm.va_copy(i8* %dest, i8* %src)
 125
 126 @second_list = global %va_list zeroinitializer
 127
 128 define void @test_va_copy() {
 129 ; CHECK-LABEL: test_va_copy:
 130   %srcaddr = bitcast %va_list* @var to i8*
 131   %dstaddr = bitcast %va_list* @second_list to i8*
 132   call void @llvm.va_copy(i8* %dstaddr, i8* %srcaddr)
 133
 134 ; CHECK: add x[[SRC:[0-9]+]], {{x[0-9]+}}, :lo12:var
 135
 136 ; CHECK: ldp [[BLOCK:q[0-9]+]], [[BLOCK:q[0-9]+]], [x[[SRC]]]
 137 ; CHECK: add x[[DST:[0-9]+]], {{x[0-9]+}}, :lo12:second_list
 138 ; CHECK: stp [[BLOCK:q[0-9]+]], [[BLOCK:q[0-9]+]], [x[[DST]]]
 139   ret void
 140 ; CHECK: ret
 141 }