test/CodeGen/AArch64/jump-table.ll

   1 ; RUN: llc -verify-machineinstrs < %s -mtriple=aarch64-none-linux-gnu | FileCheck %s
   2 ; RUN: llc -code-model=large -verify-machineinstrs < %s -mtriple=aarch64-none-linux-gnu | FileCheck --check-prefix=CHECK-LARGE %s
   3 ; RUN: llc -mtriple=aarch64-none-linux-gnu -verify-machineinstrs -relocation-model=pic <%s | FileCheck --check-prefix=CHECK-PIC %s
   4 ; RUN: llc -verify-machineinstrs -o - %s -mtriple=arm64-none-linux-gnu | FileCheck %s
   5 ; RUN: llc -code-model=large -verify-machineinstrs -o - %s -mtriple=arm64-none-linux-gnu | FileCheck --check-prefix=CHECK-LARGE %s
   6 ; RUN: llc -mtriple=arm64-none-linux-gnu -verify-machineinstrs -relocation-model=pic -o - %s | FileCheck --check-prefix=CHECK-PIC %s
   7
   8 define i32 @test_jumptable(i32 %in) {
   9 ; CHECK: test_jumptable
  10
  11   switch i32 %in, label %def [
  12     i32 0, label %lbl1
  13     i32 1, label %lbl2
  14     i32 2, label %lbl3
  15     i32 4, label %lbl4
  16   ]
  17 ; CHECK: adrp [[JTPAGE:x[0-9]+]], .LJTI0_0
  18 ; CHECK: add x[[JT:[0-9]+]], [[JTPAGE]], {{#?}}:lo12:.LJTI0_0
  19 ; CHECK: ldr [[DEST:x[0-9]+]], [x[[JT]], {{x[0-9]+}}, lsl #3]
  20 ; CHECK: br [[DEST]]
  21
  22 ; CHECK-LARGE: movz x[[JTADDR:[0-9]+]], #:abs_g3:.LJTI0_0
  23 ; CHECK-LARGE: movk x[[JTADDR]], #:abs_g2_nc:.LJTI0_0
  24 ; CHECK-LARGE: movk x[[JTADDR]], #:abs_g1_nc:.LJTI0_0
  25 ; CHECK-LARGE: movk x[[JTADDR]], #:abs_g0_nc:.LJTI0_0
  26 ; CHECK-LARGE: ldr [[DEST:x[0-9]+]], [x[[JTADDR]], {{x[0-9]+}}, lsl #3]
  27 ; CHECK-LARGE: br [[DEST]]
  28
  29 ; CHECK-PIC: adrp [[JTPAGE:x[0-9]+]], .LJTI0_0
  30 ; CHECK-PIC: add x[[JT:[0-9]+]], [[JTPAGE]], {{#?}}:lo12:.LJTI0_0
  31 ; CHECK-PIC: ldrsw [[DEST:x[0-9]+]], [x[[JT]], {{x[0-9]+}}, lsl #2]
  32 ; CHECK-PIC: add [[TABLE:x[0-9]+]], [[DEST]], x[[JT]]
  33 ; CHECK-PIC: br [[TABLE]]
  34
  35 def:
  36   ret i32 0
  37
  38 lbl1:
  39   ret i32 1
  40
  41 lbl2:
  42   ret i32 2
  43
  44 lbl3:
  45   ret i32 4
  46
  47 lbl4:
  48   ret i32 8
  49
  50 }
  51
  52 ; CHECK: .rodata
  53
  54 ; CHECK: .LJTI0_0:
  55 ; CHECK-NEXT: .xword
  56 ; CHECK-NEXT: .xword
  57 ; CHECK-NEXT: .xword
  58 ; CHECK-NEXT: .xword
  59 ; CHECK-NEXT: .xword
  60
  61 ; CHECK-PIC-NOT: .data_region
  62 ; CHECK-PIC: .LJTI0_0:
  63 ; CHECK-PIC-NEXT: .word
  64 ; CHECK-PIC-NEXT: .word
  65 ; CHECK-PIC-NEXT: .word
  66 ; CHECK-PIC-NEXT: .word
  67 ; CHECK-PIC-NEXT: .word
  68 ; CHECK-PIC-NOT: .end_data_region