test/CodeGen/AArch64/jump-table.ll

   1 ; RUN: llc -verify-machineinstrs < %s -mtriple=aarch64-none-linux-gnu | FileCheck %s
   2 ; RUN: llc -verify-machineinstrs < %s -mtriple=aarch64-none-linux-gnu -filetype=obj | llvm-readobj -r | FileCheck %s -check-prefix=CHECK-ELF
   3
   4 define i32 @test_jumptable(i32 %in) {
   5 ; CHECK: test_jumptable
   6
   7   switch i32 %in, label %def [
   8     i32 0, label %lbl1
   9     i32 1, label %lbl2
  10     i32 2, label %lbl3
  11     i32 4, label %lbl4
  12   ]
  13 ; CHECK: adrp [[JTPAGE:x[0-9]+]], .LJTI0_0
  14 ; CHECK: add x[[JT:[0-9]+]], [[JTPAGE]], #:lo12:.LJTI0_0
  15 ; CHECK: ldr [[DEST:x[0-9]+]], [x[[JT]], {{x[0-9]+}}, lsl #3]
  16 ; CHECK: br [[DEST]]
  17
  18 def:
  19   ret i32 0
  20
  21 lbl1:
  22   ret i32 1
  23
  24 lbl2:
  25   ret i32 2
  26
  27 lbl3:
  28   ret i32 4
  29
  30 lbl4:
  31   ret i32 8
  32
  33 }
  34
  35 ; CHECK: .rodata
  36
  37 ; CHECK: .LJTI0_0:
  38 ; CHECK-NEXT: .xword
  39 ; CHECK-NEXT: .xword
  40 ; CHECK-NEXT: .xword
  41 ; CHECK-NEXT: .xword
  42 ; CHECK-NEXT: .xword
  43
  44 ; ELF tests:
  45
  46 ; First make sure we get a page/lo12 pair in .text to pick up the jump-table
  47
  48 ; CHECK-ELF:      Relocations [
  49 ; CHECK-ELF:        Section ({{[0-9]+}}) .text {
  50 ; CHECK-ELF-NEXT:     0x{{[0-9,A-F]+}} R_AARCH64_ADR_PREL_PG_HI21 .rodata
  51 ; CHECK-ELF-NEXT:     0x{{[0-9,A-F]+}} R_AARCH64_ADD_ABS_LO12_NC .rodata
  52 ; CHECK-ELF:        }
  53
  54 ; Also check the targets in .rodata are relocated
  55 ; CHECK-ELF:        Section ({{[0-9]+}}) .rodata {
  56 ; CHECK-ELF-NEXT:     0x{{[0-9,A-F]+}} R_AARCH64_ABS64 .text
  57 ; CHECK-ELF:        }
  58 ; CHECK-ELF:      ]