llvm/test/CodeGen/AArch64/machine-outliner-retaddr-sign-regsave.mir

   1 # RUN: llc -mtriple=aarch64 -run-pass=prologepilog -run-pass=machine-outliner \
   2 # RUN: -verify-machineinstrs -run-pass=aarch64-ptrauth %s -o - | FileCheck %s
   3
   4 # Check that we save LR to a callee-saved register when possible.
   5 # foo() should use a callee-saved register. However, bar() should not.
   6 --- |
   7
   8   define void @foo() #0 {
   9     ret void
  10   }
  11
  12   define void @bar() #1 {
  13     ret void
  14   }
  15
  16   attributes #0 = { nounwind         "sign-return-address"="non-leaf" "sign-return-address-key"="b_key" minsize noinline noredzone "frame-pointer"="all" }
  17   attributes #1 = { nounwind uwtable "sign-return-address"="non-leaf" "sign-return-address-key"="b_key" minsize noinline noredzone "frame-pointer"="all" }
  18 ...
  19 ---
  20 # CHECK-LABEL:   name:            foo
  21 # CHECK:         bb.0:
  22 # CHECK:            frame-setup EMITBKEY
  23 # CHECK-NEXT:       frame-setup PACIBSP
  24 # CHECK-NOT:        frame-setup CFI_INSTRUCTION negate_ra_sign_state
  25 # CHECK:        bb.1:
  26 # CHECK:            BL @[[OUTLINED_FUNCTION:OUTLINED_FUNCTION_[0-9]+]]
  27 # CHECK:        bb.2:
  28 # CHECK:            BL @[[OUTLINED_FUNCTION]]
  29 # CHECK:        bb.3:
  30 # CHECK:            BL @[[OUTLINED_FUNCTION]]
  31 # CHECK:        bb.4:
  32 # CHECK:            BL @[[OUTLINED_FUNCTION]]
  33 # CHECK:        bb.5:
  34 # CHECK:            frame-destroy AUTIBSP
  35 # CHECK-NEXT:       RET
  36 name:            foo
  37 tracksRegLiveness: true
  38 fixedStack:
  39 body:             |
  40   bb.0:
  41     $x25 = ORRXri $xzr, 1
  42     $lr = ORRXri $xzr, 1
  43   bb.1:
  44     liveins: $lr, $w9
  45     $w9 = ORRWri $wzr, 1
  46     $w9 = ORRWri $wzr, 1
  47     $w9 = ORRWri $wzr, 1
  48     $w9 = ORRWri $wzr, 1
  49     $w9 = ORRWri $wzr, 1
  50     $w9 = ORRWri $wzr, 2
  51   bb.2:
  52     liveins: $lr, $w9
  53     $w9 = ORRWri $wzr, 1
  54     $w9 = ORRWri $wzr, 1
  55     $w9 = ORRWri $wzr, 1
  56     $w9 = ORRWri $wzr, 1
  57     $w9 = ORRWri $wzr, 1
  58     $w9 = ORRWri $wzr, 2
  59   bb.3:
  60     liveins: $lr, $w9
  61     $w9 = ORRWri $wzr, 1
  62     $w9 = ORRWri $wzr, 1
  63     $w9 = ORRWri $wzr, 1
  64     $w9 = ORRWri $wzr, 1
  65     $w9 = ORRWri $wzr, 1
  66     $w9 = ORRWri $wzr, 2
  67   bb.4:
  68     liveins: $lr, $w9
  69     $w9 = ORRWri $wzr, 1
  70     $w9 = ORRWri $wzr, 1
  71     $w9 = ORRWri $wzr, 1
  72     $w9 = ORRWri $wzr, 1
  73     $w9 = ORRWri $wzr, 1
  74     $w9 = ORRWri $wzr, 2
  75   bb.5:
  76     liveins: $w9
  77     RET undef $lr
  78
  79 ...
  80 ---
  81 # CHECK:         name:            bar
  82 # CHECK:          bb.0:
  83 # CHECK:            frame-setup EMITBKEY
  84 # CHECK-NEXT:       frame-setup PACIBSP implicit-def $lr, implicit $lr, implicit $sp
  85 # CHECK:            frame-setup CFI_INSTRUCTION negate_ra_sign_state
  86 # CHECK-NEXT:       frame-setup CFI_INSTRUCTION
  87 # CHECK-NOT:        OUTLINED_FUNCTION_
  88 # CHECK:          bb.1:
  89 # CHECK-NOT:        OUTLINED_FUNCTION_
  90 # CHECK:          bb.2:
  91 # CHECK-NOT:        OUTLINED_FUNCTION_
  92 # CHECK:          bb.3:
  93 # CHECK-NOT:        OUTLINED_FUNCTION_
  94 # CHECK:            RET
  95 name:            bar
  96 tracksRegLiveness: true
  97 body:             |
  98   bb.0:
  99     liveins: $lr, $x0, $x1, $x2, $x3, $x4, $x5, $x6, $x7, $x8, $x9, $x10, $x11, $x12, $x13, $x14, $x15, $x19, $x20, $x21, $x22, $x23, $x20, $x21, $x22, $x23, $x24, $x25, $x26, $x27, $x28
 100     $w10 = ORRWri $wzr, 1
 101     $w10 = ORRWri $wzr, 1
 102     $w10 = ORRWri $wzr, 1
 103     $w10 = ORRWri $wzr, 1
 104     $w10 = ORRWri $wzr, 1
 105     $w12 = ORRWri $wzr, 2
 106   bb.1:
 107     liveins: $lr, $x0, $x1, $x2, $x3, $x4, $x5, $x6, $x7, $x8, $x9, $x10, $x11, $x12, $x13, $x14, $x15, $x19, $x20, $x21, $x22, $x23, $x20, $x21, $x22, $x23, $x24, $x25, $x26, $x27, $x28
 108     $w10 = ORRWri $wzr, 1
 109     $w10 = ORRWri $wzr, 1
 110     $w10 = ORRWri $wzr, 1
 111     $w10 = ORRWri $wzr, 1
 112     $w10 = ORRWri $wzr, 1
 113     $w12 = ORRWri $wzr, 2
 114   bb.2:
 115     liveins: $lr, $x0, $x1, $x2, $x3, $x4, $x5, $x6, $x7, $x8, $x9, $x10, $x11, $x12, $x13, $x14, $x15, $x19, $x20, $x21, $x22, $x23, $x20, $x21, $x22, $x23, $x24, $x25, $x26, $x27, $x28
 116     $w10 = ORRWri $wzr, 1
 117     $w10 = ORRWri $wzr, 1
 118     $w10 = ORRWri $wzr, 1
 119     $w10 = ORRWri $wzr, 1
 120     $w10 = ORRWri $wzr, 1
 121     $w12 = ORRWri $wzr, 2
 122   bb.3:
 123     liveins: $lr, $x0, $x1, $x2, $x3, $x4, $x5, $x6, $x7, $x8, $x9, $x10, $x11, $x12, $x13, $x14, $x15, $x19, $x20, $x21, $x22, $x23, $x20, $x21, $x22, $x23, $x24, $x25, $x26, $x27, $x28
 124     RET undef $lr
 125
 126 # CHECK:         name:            [[OUTLINED_FUNCTION]]
 127 # CHECK:         body:
 128 # CHECK-NEXT:     bb.0:
 129 # CHECK-NOT:        frame-setup EMITBKEY
 130 # CHECK-NOT:        frame-setup PACI{{[A,B]}}SP
 131 # CHECK-NOT:        frame-setup CFI_INSTRUCTION negate_ra_sign_state
 132 # CHECK-NOT:        frame-destroy AUTI{{[A,B]}}SP