test/CodeGen/MIR/X86/jump-table-info.mir

   1 # RUN: llc -march=x86-64 -start-after branch-folder -stop-after branch-folder -o /dev/null %s | FileCheck %s
   2 # This test ensures that the MIR parser parses the jump table info and jump
   3 # table operands correctly.
   4
   5 --- |
   6
   7   define i32 @test_jumptable(i32 %in) {
   8   entry:
   9     switch i32 %in, label %def [
  10       i32 0, label %lbl1
  11       i32 1, label %lbl2
  12       i32 2, label %lbl3
  13       i32 3, label %lbl4
  14     ]
  15
  16   def:
  17     ret i32 0
  18
  19   lbl1:
  20     ret i32 1
  21
  22   lbl2:
  23     ret i32 2
  24
  25   lbl3:
  26     ret i32 4
  27
  28   lbl4:
  29     ret i32 8
  30   }
  31
  32   define i32 @test_jumptable2(i32 %in) {
  33   entry:
  34     switch i32 %in, label %def [
  35       i32 0, label %lbl1
  36       i32 1, label %lbl2
  37       i32 2, label %lbl3
  38       i32 3, label %lbl4
  39     ]
  40
  41   def:
  42     ret i32 0
  43
  44   lbl1:
  45     ret i32 1
  46
  47   lbl2:
  48     ret i32 2
  49
  50   lbl3:
  51     ret i32 4
  52
  53   lbl4:
  54     ret i32 8
  55   }
  56
  57 ...
  58 ---
  59 name:            test_jumptable
  60 # CHECK:      jumpTable:
  61 # CHECK-NEXT: kind: label-difference32
  62 # CHECK-NEXT: entries:
  63 # CHECK-NEXT: - id: 0
  64 # CHECK-NEXT: blocks: [ '%bb.3.lbl1', '%bb.4.lbl2', '%bb.5.lbl3', '%bb.6.lbl4' ]
  65 # CHECK_NEXT: body:
  66 jumpTable:
  67   kind:          label-difference32
  68   entries:
  69     - id:        0
  70       blocks:    [ '%bb.3.lbl1', '%bb.4.lbl2', '%bb.5.lbl3', '%bb.6.lbl4' ]
  71 body: |
  72   bb.0.entry:
  73     successors: %bb.2.def, %bb.1.entry
  74
  75     %eax = MOV32rr %edi, implicit-def %rax
  76     CMP32ri8 %edi, 3, implicit-def %eflags
  77     JA_1 %bb.2.def, implicit %eflags
  78
  79   bb.1.entry:
  80     successors: %bb.3.lbl1, %bb.4.lbl2, %bb.5.lbl3, %bb.6.lbl4
  81     ; CHECK: %rcx = LEA64r %rip, 1, _, %jump-table.0, _
  82     %rcx = LEA64r %rip, 1, _, %jump-table.0, _
  83     %rax = MOVSX64rm32 %rcx, 4, %rax, 0, _
  84     %rax = ADD64rr %rax, %rcx, implicit-def %eflags
  85     JMP64r %rax
  86
  87   bb.2.def:
  88     %eax = MOV32r0 implicit-def %eflags
  89     RETQ %eax
  90
  91   bb.3.lbl1:
  92     %eax = MOV32ri 1
  93     RETQ %eax
  94
  95   bb.4.lbl2:
  96     %eax = MOV32ri 2
  97     RETQ %eax
  98
  99   bb.5.lbl3:
 100     %eax = MOV32ri 4
 101     RETQ %eax
 102
 103   bb.6.lbl4:
 104     %eax = MOV32ri 8
 105     RETQ %eax
 106 ...
 107 ---
 108 name:            test_jumptable2
 109 jumpTable:
 110   kind:          label-difference32
 111   entries:
 112     - id:        1
 113       blocks:    [ '%bb.3.lbl1', '%bb.4.lbl2', '%bb.5.lbl3', '%bb.6.lbl4' ]
 114 body: |
 115   bb.0.entry:
 116     successors: %bb.2.def, %bb.1.entry
 117
 118     %eax = MOV32rr %edi, implicit-def %rax
 119     CMP32ri8 %edi, 3, implicit-def %eflags
 120     JA_1 %bb.2.def, implicit %eflags
 121
 122   bb.1.entry:
 123     successors: %bb.3.lbl1, %bb.4.lbl2, %bb.5.lbl3, %bb.6.lbl4
 124     ; Verify that the printer will use an id of 0 for this jump table:
 125     ; CHECK: %rcx = LEA64r %rip, 1, _, %jump-table.0, _
 126     %rcx = LEA64r %rip, 1, _, %jump-table.1, _
 127     %rax = MOVSX64rm32 %rcx, 4, %rax, 0, _
 128     %rax = ADD64rr %rax, %rcx, implicit-def %eflags
 129     JMP64r %rax
 130
 131   bb.2.def:
 132     %eax = MOV32r0 implicit-def %eflags
 133     RETQ %eax
 134
 135   bb.3.lbl1:
 136     %eax = MOV32ri 1
 137     RETQ %eax
 138
 139   bb.4.lbl2:
 140     %eax = MOV32ri 2
 141     RETQ %eax
 142
 143   bb.5.lbl3:
 144     %eax = MOV32ri 4
 145     RETQ %eax
 146
 147   bb.6.lbl4:
 148     %eax = MOV32ri 8
 149     RETQ %eax
 150 ...