test/CodeGen/AArch64/machine-outliner-regsave.mir

   1 # RUN: llc -mtriple=aarch64-apple-darwin -run-pass=prologepilog \
   2 # RUN: -run-pass=machine-outliner -verify-machineinstrs %s -o - | FileCheck %s
   3 # Check that we save LR to a callee-saved register when possible.
   4 # foo() should use a callee-saved register. However, bar() should not.
   5 --- |
   6
   7   define void @foo() #0 {
   8     ret void
   9   }
  10
  11   define void @bar() #0 {
  12     ret void
  13   }
  14
  15   attributes #0 = { minsize noinline noredzone "no-frame-pointer-elim"="true" }
  16 ...
  17 ---
  18 # Make sure that when we outline and a register is available, we
  19 # use it to save + restore LR instead of SP.
  20 # Also make sure that we can call functions that require no save as the same
  21 # outlined function.
  22 # CHECK: name:            foo
  23 # CHECK-DAG: bb.1
  24 # CHECK-DAG: $x[[REG:[0-9]+]] = ORRXrs $xzr, $lr, 0
  25 # CHECK-NEXT: BL
  26 # CHECK-NEXT: $lr = ORRXrs $xzr, $x[[REG]], 0
  27 # CHECK-DAG: bb.2
  28 # CHECK-DAG: $x[[REG]] = ORRXrs $xzr, $lr, 0
  29 # CHECK-NEXT: BL
  30 # CHECK-NEXT: $lr = ORRXrs $xzr, $x[[REG]], 0
  31 # CHECK-DAG: bb.3
  32 # CHECK-DAG: $x[[REG]] = ORRXrs $xzr, $lr, 0
  33 # CHECK-NEXT: BL
  34 # CHECK-NEXT: $lr = ORRXrs $xzr, $x[[REG]], 0
  35 # CHECK-DAG: bb.4
  36 # CHECK-NOT: $x[[REG]] = ORRXrs $xzr, $lr, 0
  37 # CHECK-DAG: BL
  38 name:            foo
  39 tracksRegLiveness: true
  40 fixedStack:
  41 body:             |
  42   bb.0:
  43     $x25 = ORRXri $xzr, 1
  44     $lr = ORRXri $xzr, 1
  45   bb.1:
  46     liveins: $lr, $w9
  47     $w9 = ORRWri $wzr, 1
  48     $w9 = ORRWri $wzr, 1
  49     $w9 = ORRWri $wzr, 1
  50     $w9 = ORRWri $wzr, 1
  51     $w9 = ORRWri $wzr, 1
  52     $w9 = ORRWri $wzr, 2
  53   bb.2:
  54     liveins: $lr, $w9
  55     $w9 = ORRWri $wzr, 1
  56     $w9 = ORRWri $wzr, 1
  57     $w9 = ORRWri $wzr, 1
  58     $w9 = ORRWri $wzr, 1
  59     $w9 = ORRWri $wzr, 1
  60     $w9 = ORRWri $wzr, 2
  61   bb.3:
  62     liveins: $lr, $w9
  63     $w9 = ORRWri $wzr, 1
  64     $w9 = ORRWri $wzr, 1
  65     $w9 = ORRWri $wzr, 1
  66     $w9 = ORRWri $wzr, 1
  67     $w9 = ORRWri $wzr, 1
  68     $w9 = ORRWri $wzr, 2
  69   bb.4:
  70     liveins: $lr, $w9
  71     $w9 = ORRWri $wzr, 1
  72     $w9 = ORRWri $wzr, 1
  73     $w9 = ORRWri $wzr, 1
  74     $w9 = ORRWri $wzr, 1
  75     $w9 = ORRWri $wzr, 1
  76     $w9 = ORRWri $wzr, 2
  77   bb.5:
  78     liveins: $w9
  79     RET undef $lr
  80
  81 ...
  82 ---
  83 # Convoluted case that shows that we'll still save to the stack when there are
  84 # no approprate registers available.
  85 # The live-in lists do not contain x16 or x17 since including them would cause
  86 # nothing to be outlined.
  87 # They also deliberately don't contain x18 to show that on Darwin we won't store
  88 # to that.
  89 # CHECK-LABEL: name:            bar
  90 # CHECK: early-clobber $sp = STRXpre $lr, $sp, -16
  91 # CHECK-NEXT: BL
  92 # CHECK-DAG: early-clobber $sp, $lr = LDRXpost $sp, 16
  93 # CHECK: early-clobber $sp = STRXpre $lr, $sp, -16
  94 # CHECK-NEXT: BL
  95 # CHECK-DAG: early-clobber $sp, $lr = LDRXpost $sp, 16
  96 # CHECK: early-clobber $sp = STRXpre $lr, $sp, -16
  97 # CHECK-NEXT: BL
  98 # CHECK-NEXT: early-clobber $sp, $lr = LDRXpost $sp, 16
  99 name:            bar
 100 tracksRegLiveness: true
 101 body:             |
 102   bb.0:
 103     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
 104     $w10 = ORRWri $wzr, 1
 105     $w10 = ORRWri $wzr, 1
 106     $w10 = ORRWri $wzr, 1
 107     $w10 = ORRWri $wzr, 1
 108     $w10 = ORRWri $wzr, 1
 109     $w12 = ORRWri $wzr, 2
 110   bb.1:
 111     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
 112     $w10 = ORRWri $wzr, 1
 113     $w10 = ORRWri $wzr, 1
 114     $w10 = ORRWri $wzr, 1
 115     $w10 = ORRWri $wzr, 1
 116     $w10 = ORRWri $wzr, 1
 117     $w12 = ORRWri $wzr, 2
 118   bb.2:
 119     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
 120     $w10 = ORRWri $wzr, 1
 121     $w10 = ORRWri $wzr, 1
 122     $w10 = ORRWri $wzr, 1
 123     $w10 = ORRWri $wzr, 1
 124     $w10 = ORRWri $wzr, 1
 125     $w12 = ORRWri $wzr, 2
 126   bb.3:
 127     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
 128     RET undef $lr