llvm/test/CodeGen/X86/switch-jmp-edge-split.mir

   1 # NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py
   2 # RUN: llc -mtriple=x86_64-- -run-pass=machine-sink -verify-machineinstrs  -o - %s | FileCheck %s
   3
   4 ---
   5 name:            func_no_pic
   6 tracksRegLiveness: true
   7 jumpTable:
   8   kind:            block-address
   9   entries:
  10     - id:              0
  11       blocks:          [ '%bb.4', '%bb.1', '%bb.2', '%bb.3' ]
  12 body:             |
  13   ; CHECK-LABEL: name: func_no_pic
  14   ; CHECK: bb.0:
  15   ; CHECK-NEXT:   successors: %bb.5(0x20000000), %bb.1(0x20000000), %bb.2(0x20000000), %bb.3(0x20000000)
  16   ; CHECK-NEXT:   liveins: $edi
  17   ; CHECK-NEXT: {{  $}}
  18   ; CHECK-NEXT:   [[COPY:%[0-9]+]]:gr64_nosp = COPY $rdi
  19   ; CHECK-NEXT:   JMP64m $noreg, 8, [[COPY]], %jump-table.0, $noreg :: (load (s64) from jump-table)
  20   ; CHECK-NEXT: {{  $}}
  21   ; CHECK-NEXT: bb.5:
  22   ; CHECK-NEXT:   successors: %bb.4(0x80000000)
  23   ; CHECK-NEXT: {{  $}}
  24   ; CHECK-NEXT:   [[MOV32ri:%[0-9]+]]:gr32 = MOV32ri 42
  25   ; CHECK-NEXT:   JMP_1 %bb.4
  26   ; CHECK-NEXT: {{  $}}
  27   ; CHECK-NEXT: bb.1:
  28   ; CHECK-NEXT:   successors: %bb.4(0x80000000)
  29   ; CHECK-NEXT: {{  $}}
  30   ; CHECK-NEXT:   [[MOV32ri1:%[0-9]+]]:gr32 = MOV32ri 6
  31   ; CHECK-NEXT:   JMP_1 %bb.4
  32   ; CHECK-NEXT: {{  $}}
  33   ; CHECK-NEXT: bb.2:
  34   ; CHECK-NEXT:   successors: %bb.4(0x80000000)
  35   ; CHECK-NEXT: {{  $}}
  36   ; CHECK-NEXT:   [[MOV32ri2:%[0-9]+]]:gr32 = MOV32ri 1
  37   ; CHECK-NEXT:   JMP_1 %bb.4
  38   ; CHECK-NEXT: {{  $}}
  39   ; CHECK-NEXT: bb.3:
  40   ; CHECK-NEXT:   successors: %bb.4(0x80000000)
  41   ; CHECK-NEXT: {{  $}}
  42   ; CHECK-NEXT:   [[MOV32ri3:%[0-9]+]]:gr32 = MOV32ri 92
  43   ; CHECK-NEXT:   JMP_1 %bb.4
  44   ; CHECK-NEXT: {{  $}}
  45   ; CHECK-NEXT: bb.4:
  46   ; CHECK-NEXT:   [[PHI:%[0-9]+]]:gr32 = PHI [[MOV32ri]], %bb.5, [[MOV32ri1]], %bb.1, [[MOV32ri2]], %bb.2, [[MOV32ri3]], %bb.3
  47   ; CHECK-NEXT:   $eax = COPY [[PHI]]
  48   ; CHECK-NEXT:   RET 0
  49   bb.0:
  50     successors: %bb.4, %bb.1, %bb.2, %bb.3
  51     liveins: $edi
  52
  53     %0:gr64_nosp = COPY $rdi
  54     ; MachineSink should split the critical edges and sink the next two
  55     ; insns out of the switch-jump header into new less frequently executed
  56     ; blocks.
  57     %1:gr32 = MOV32ri 42
  58     JMP64m $noreg, 8, %0, %jump-table.0, $noreg :: (load (s64) from jump-table)
  59
  60   bb.1:
  61     %2:gr32 = MOV32ri 6
  62     JMP_1 %bb.4
  63
  64   bb.2:
  65     %3:gr32 = MOV32ri 1
  66     JMP_1 %bb.4
  67
  68   bb.3:
  69     %4:gr32 = MOV32ri 92
  70     JMP_1 %bb.4
  71
  72   bb.4:
  73     %5:gr32 = PHI %1, %bb.0, %2, %bb.1, %3, %bb.2, %4, %bb.3
  74     $eax = COPY %5
  75     RET 0
  76
  77 ...
  78 ---
  79 name:            func_pic
  80 tracksRegLiveness: true
  81 jumpTable:
  82   kind:            label-difference32
  83   entries:
  84     - id:              0
  85       blocks:          [ '%bb.4', '%bb.1', '%bb.2', '%bb.3' ]
  86 body:             |
  87   ; CHECK-LABEL: name: func_pic
  88   ; CHECK: bb.0:
  89   ; CHECK-NEXT:   successors: %bb.5(0x20000000), %bb.1(0x20000000), %bb.2(0x20000000), %bb.3(0x20000000)
  90   ; CHECK-NEXT:   liveins: $edi
  91   ; CHECK-NEXT: {{  $}}
  92   ; CHECK-NEXT:   [[COPY:%[0-9]+]]:gr64_nosp = COPY $rdi
  93   ; CHECK-NEXT:   [[LEA64r:%[0-9]+]]:gr64 = LEA64r $rip, 1, $noreg, %jump-table.0, $noreg
  94   ; CHECK-NEXT:   [[MOVSX64rm32_:%[0-9]+]]:gr64 = MOVSX64rm32 [[LEA64r]], 4, [[COPY]], 0, $noreg :: (load (s32) from jump-table)
  95   ; CHECK-NEXT:   [[ADD64rr:%[0-9]+]]:gr64 = ADD64rr [[MOVSX64rm32_]], [[LEA64r]], implicit-def dead $eflags
  96   ; CHECK-NEXT:   JMP64r killed [[ADD64rr]]
  97   ; CHECK-NEXT: {{  $}}
  98   ; CHECK-NEXT: bb.5:
  99   ; CHECK-NEXT:   successors: %bb.4(0x80000000)
 100   ; CHECK-NEXT: {{  $}}
 101   ; CHECK-NEXT:   [[MOV32ri:%[0-9]+]]:gr32 = MOV32ri 42
 102   ; CHECK-NEXT:   JMP_1 %bb.4
 103   ; CHECK-NEXT: {{  $}}
 104   ; CHECK-NEXT: bb.1:
 105   ; CHECK-NEXT:   successors: %bb.4(0x80000000)
 106   ; CHECK-NEXT: {{  $}}
 107   ; CHECK-NEXT:   [[MOV32ri1:%[0-9]+]]:gr32 = MOV32ri 6
 108   ; CHECK-NEXT:   JMP_1 %bb.4
 109   ; CHECK-NEXT: {{  $}}
 110   ; CHECK-NEXT: bb.2:
 111   ; CHECK-NEXT:   successors: %bb.4(0x80000000)
 112   ; CHECK-NEXT: {{  $}}
 113   ; CHECK-NEXT:   [[MOV32ri2:%[0-9]+]]:gr32 = MOV32ri 1
 114   ; CHECK-NEXT:   JMP_1 %bb.4
 115   ; CHECK-NEXT: {{  $}}
 116   ; CHECK-NEXT: bb.3:
 117   ; CHECK-NEXT:   successors: %bb.4(0x80000000)
 118   ; CHECK-NEXT: {{  $}}
 119   ; CHECK-NEXT:   [[MOV32ri3:%[0-9]+]]:gr32 = MOV32ri 92
 120   ; CHECK-NEXT:   JMP_1 %bb.4
 121   ; CHECK-NEXT: {{  $}}
 122   ; CHECK-NEXT: bb.4:
 123   ; CHECK-NEXT:   [[PHI:%[0-9]+]]:gr32 = PHI [[MOV32ri]], %bb.5, [[MOV32ri1]], %bb.1, [[MOV32ri2]], %bb.2, [[MOV32ri3]], %bb.3
 124   ; CHECK-NEXT:   $eax = COPY [[PHI]]
 125   ; CHECK-NEXT:   RET 0
 126   bb.0:
 127     successors: %bb.4, %bb.1, %bb.2, %bb.3
 128     liveins: $edi
 129
 130     %0:gr64_nosp = COPY $rdi
 131     ; MachineSink should split the critical edges and sink the next two
 132     ; insns out of the switch-jump header into new less frequently executed
 133     ; blocks.
 134     %1:gr32 = MOV32ri 42
 135     %2:gr64 = LEA64r $rip, 1, $noreg, %jump-table.0, $noreg
 136     %3:gr64 = MOVSX64rm32 %2, 4, %0, 0, $noreg :: (load (s32) from jump-table)
 137     %4:gr64 = ADD64rr %3, %2, implicit-def dead $eflags
 138     JMP64r killed %4
 139
 140   bb.1:
 141     %5:gr32 = MOV32ri 6
 142     JMP_1 %bb.4
 143
 144   bb.2:
 145     %6:gr32 = MOV32ri 1
 146     JMP_1 %bb.4
 147
 148   bb.3:
 149     %7:gr32 = MOV32ri 92
 150     JMP_1 %bb.4
 151
 152   bb.4:
 153     %8:gr32 = PHI %1, %bb.0, %5, %bb.1, %6, %bb.2, %7, %bb.3
 154     $eax = COPY %8
 155     RET 0
 156
 157 ...
 158 ---
 159 name:            multiple_jump_table_users
 160 tracksRegLiveness: true
 161 jumpTable:
 162   kind:            block-address
 163   entries:
 164     - id:              0
 165       blocks:          [ '%bb.6', '%bb.3', '%bb.4', '%bb.5' ]
 166 body:             |
 167   ; CHECK-LABEL: name: multiple_jump_table_users
 168   ; CHECK: bb.0:
 169   ; CHECK-NEXT:   successors: %bb.1(0x40000000), %bb.2(0x40000000)
 170   ; CHECK-NEXT:   liveins: $edi
 171   ; CHECK-NEXT: {{  $}}
 172   ; CHECK-NEXT:   [[COPY:%[0-9]+]]:gr64_nosp = COPY $rdi
 173   ; CHECK-NEXT:   CMP64ri8 [[COPY]], 42, implicit-def $eflags
 174   ; CHECK-NEXT:   JCC_1 %bb.1, 15, implicit $eflags
 175   ; CHECK-NEXT:   JMP_1 %bb.2
 176   ; CHECK-NEXT: {{  $}}
 177   ; CHECK-NEXT: bb.1:
 178   ; CHECK-NEXT:   successors: %bb.6(0x20000000), %bb.3(0x20000000), %bb.4(0x20000000), %bb.5(0x20000000)
 179   ; CHECK-NEXT: {{  $}}
 180   ; CHECK-NEXT:   INLINEASM &"", 1 /* sideeffect attdialect */
 181   ; CHECK-NEXT:   [[MOV32ri:%[0-9]+]]:gr32 = MOV32ri 13
 182   ; CHECK-NEXT:   JMP64m $noreg, 8, [[COPY]], %jump-table.0, $noreg :: (load (s64) from jump-table)
 183   ; CHECK-NEXT: {{  $}}
 184   ; CHECK-NEXT: bb.2:
 185   ; CHECK-NEXT:   successors: %bb.6(0x20000000), %bb.3(0x20000000), %bb.4(0x20000000), %bb.5(0x20000000)
 186   ; CHECK-NEXT: {{  $}}
 187   ; CHECK-NEXT:   INLINEASM &"", 1 /* sideeffect attdialect */
 188   ; CHECK-NEXT:   [[MOV32ri1:%[0-9]+]]:gr32 = MOV32ri 42
 189   ; CHECK-NEXT:   JMP64m $noreg, 8, [[COPY]], %jump-table.0, $noreg :: (load (s64) from jump-table)
 190   ; CHECK-NEXT: {{  $}}
 191   ; CHECK-NEXT: bb.3:
 192   ; CHECK-NEXT:   successors: %bb.6(0x80000000)
 193   ; CHECK-NEXT: {{  $}}
 194   ; CHECK-NEXT:   [[MOV32ri2:%[0-9]+]]:gr32 = MOV32ri 6
 195   ; CHECK-NEXT:   JMP_1 %bb.6
 196   ; CHECK-NEXT: {{  $}}
 197   ; CHECK-NEXT: bb.4:
 198   ; CHECK-NEXT:   successors: %bb.6(0x80000000)
 199   ; CHECK-NEXT: {{  $}}
 200   ; CHECK-NEXT:   [[MOV32ri3:%[0-9]+]]:gr32 = MOV32ri 1
 201   ; CHECK-NEXT:   JMP_1 %bb.6
 202   ; CHECK-NEXT: {{  $}}
 203   ; CHECK-NEXT: bb.5:
 204   ; CHECK-NEXT:   successors: %bb.6(0x80000000)
 205   ; CHECK-NEXT: {{  $}}
 206   ; CHECK-NEXT:   [[MOV32ri4:%[0-9]+]]:gr32 = MOV32ri 92
 207   ; CHECK-NEXT:   JMP_1 %bb.6
 208   ; CHECK-NEXT: {{  $}}
 209   ; CHECK-NEXT: bb.6:
 210   ; CHECK-NEXT:   [[PHI:%[0-9]+]]:gr32 = PHI [[MOV32ri]], %bb.1, [[MOV32ri1]], %bb.2, [[MOV32ri2]], %bb.3, [[MOV32ri3]], %bb.4, [[MOV32ri4]], %bb.5
 211   ; CHECK-NEXT:   $eax = COPY [[PHI]]
 212   ; CHECK-NEXT:   RET 0
 213   bb.0:
 214     liveins: $edi
 215     %0:gr64_nosp = COPY $rdi
 216
 217     CMP64ri8 %0:gr64_nosp, 42, implicit-def $eflags
 218     JCC_1 %bb.1, 15, implicit $eflags
 219     JMP_1 %bb.2
 220
 221   bb.1:
 222     successors: %bb.6, %bb.3, %bb.4, %bb.5
 223     INLINEASM &"", 1 /* sideeffect attdialect */
 224     %1:gr32 = MOV32ri 13
 225     JMP64m $noreg, 8, %0, %jump-table.0, $noreg :: (load (s64) from jump-table)
 226
 227   bb.2:
 228     successors: %bb.6, %bb.3, %bb.4, %bb.5
 229     INLINEASM &"", 1 /* sideeffect attdialect */
 230     %2:gr32 = MOV32ri 42
 231     ; This is a 2d user of jump-table.0 . This case is not supported yet;
 232     ; Should not attempt edge splitting.
 233     JMP64m $noreg, 8, %0, %jump-table.0, $noreg :: (load (s64) from jump-table)
 234
 235   bb.3:
 236     %3:gr32 = MOV32ri 6
 237     JMP_1 %bb.6
 238
 239   bb.4:
 240     %4:gr32 = MOV32ri 1
 241     JMP_1 %bb.6
 242
 243   bb.5:
 244     %5:gr32 = MOV32ri 92
 245     JMP_1 %bb.6
 246
 247   bb.6:
 248     %6:gr32 = PHI %1, %bb.1, %2, %bb.2, %3, %bb.3, %4, %bb.4, %5, %bb.5
 249     $eax = COPY %6
 250     RET 0
 251 ...