llvm/test/CodeGen/RISCV/GlobalISel/instruction-select/jump-table-brjt-small-rv64.mir

   1 # NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py
   2 # RUN: llc -mtriple=riscv64 -run-pass=instruction-select %s -o - \
   3 # RUN:   -code-model=small | FileCheck %s
   4
   5 --- |
   6   define i32 @jt_test(i32 signext %in) {
   7   entry:
   8     %0 = sext i32 %in to i64
   9     switch i64 %0, label %default [
  10       i64 1, label %bb1
  11       i64 2, label %bb2
  12       i64 3, label %bb3
  13       i64 4, label %bb4
  14       i64 5, label %bb5
  15       i64 6, label %bb6
  16     ]
  17
  18   bb1:                                              ; preds = %entry
  19     ret i32 4
  20
  21   bb2:                                              ; preds = %entry
  22     ret i32 3
  23
  24   bb3:                                              ; preds = %entry
  25     ret i32 2
  26
  27   bb4:                                              ; preds = %entry
  28     ret i32 1
  29
  30   bb5:                                              ; preds = %entry
  31     ret i32 100
  32
  33   bb6:                                              ; preds = %entry
  34     ret i32 200
  35
  36   default:                                          ; preds = %entry
  37     ret i32 1000
  38   }
  39
  40 ...
  41 ---
  42 name:            jt_test
  43 legalized:       true
  44 regBankSelected: true
  45 tracksRegLiveness: true
  46 jumpTable:
  47   kind:            custom32
  48   entries:
  49     - id:              0
  50       blocks:          [ '%bb.2', '%bb.3', '%bb.4', '%bb.5', '%bb.6', '%bb.7' ]
  51 body:             |
  52   ; CHECK-LABEL: name: jt_test
  53   ; CHECK: bb.0.entry:
  54   ; CHECK-NEXT:   successors: %bb.8(0x40000000), %bb.1(0x40000000)
  55   ; CHECK-NEXT:   liveins: $x10
  56   ; CHECK-NEXT: {{  $}}
  57   ; CHECK-NEXT:   [[COPY:%[0-9]+]]:gpr = COPY $x10
  58   ; CHECK-NEXT:   [[ADDI:%[0-9]+]]:gpr = ADDI $x0, 5
  59   ; CHECK-NEXT:   [[ADDIW:%[0-9]+]]:gpr = ADDIW [[COPY]], 0
  60   ; CHECK-NEXT:   [[ADDI1:%[0-9]+]]:gpr = ADDI [[ADDIW]], -1
  61   ; CHECK-NEXT:   BLTU [[ADDI]], [[ADDI1]], %bb.8
  62   ; CHECK-NEXT: {{  $}}
  63   ; CHECK-NEXT: bb.1.entry:
  64   ; CHECK-NEXT:   successors: %bb.2(0x15555555), %bb.3(0x15555555), %bb.4(0x15555555), %bb.5(0x15555555), %bb.6(0x15555555), %bb.7(0x15555555)
  65   ; CHECK-NEXT: {{  $}}
  66   ; CHECK-NEXT:   [[LUI:%[0-9]+]]:gpr = LUI target-flags(riscv-hi) %jump-table.0
  67   ; CHECK-NEXT:   [[ADDI2:%[0-9]+]]:gpr = ADDI [[LUI]], target-flags(riscv-lo) %jump-table.0
  68   ; CHECK-NEXT:   [[SLLI:%[0-9]+]]:gpr = SLLI [[ADDI1]], 2
  69   ; CHECK-NEXT:   [[ADD:%[0-9]+]]:gpr = ADD [[ADDI2]], [[SLLI]]
  70   ; CHECK-NEXT:   [[LW:%[0-9]+]]:gprjalr = LW [[ADD]], 0 :: (load (s32) from jump-table)
  71   ; CHECK-NEXT:   PseudoBRIND [[LW]], 0
  72   ; CHECK-NEXT: {{  $}}
  73   ; CHECK-NEXT: bb.2.bb1:
  74   ; CHECK-NEXT:   [[ADDI3:%[0-9]+]]:gpr = ADDI $x0, 4
  75   ; CHECK-NEXT:   $x10 = COPY [[ADDI3]]
  76   ; CHECK-NEXT:   PseudoRET implicit $x10
  77   ; CHECK-NEXT: {{  $}}
  78   ; CHECK-NEXT: bb.3.bb2:
  79   ; CHECK-NEXT:   [[ADDI4:%[0-9]+]]:gpr = ADDI $x0, 3
  80   ; CHECK-NEXT:   $x10 = COPY [[ADDI4]]
  81   ; CHECK-NEXT:   PseudoRET implicit $x10
  82   ; CHECK-NEXT: {{  $}}
  83   ; CHECK-NEXT: bb.4.bb3:
  84   ; CHECK-NEXT:   [[ADDI5:%[0-9]+]]:gpr = ADDI $x0, 2
  85   ; CHECK-NEXT:   $x10 = COPY [[ADDI5]]
  86   ; CHECK-NEXT:   PseudoRET implicit $x10
  87   ; CHECK-NEXT: {{  $}}
  88   ; CHECK-NEXT: bb.5.bb4:
  89   ; CHECK-NEXT:   [[ADDI6:%[0-9]+]]:gpr = ADDI $x0, 1
  90   ; CHECK-NEXT:   $x10 = COPY [[ADDI6]]
  91   ; CHECK-NEXT:   PseudoRET implicit $x10
  92   ; CHECK-NEXT: {{  $}}
  93   ; CHECK-NEXT: bb.6.bb5:
  94   ; CHECK-NEXT:   [[ADDI7:%[0-9]+]]:gpr = ADDI $x0, 100
  95   ; CHECK-NEXT:   $x10 = COPY [[ADDI7]]
  96   ; CHECK-NEXT:   PseudoRET implicit $x10
  97   ; CHECK-NEXT: {{  $}}
  98   ; CHECK-NEXT: bb.7.bb6:
  99   ; CHECK-NEXT:   [[ADDI8:%[0-9]+]]:gpr = ADDI $x0, 200
 100   ; CHECK-NEXT:   $x10 = COPY [[ADDI8]]
 101   ; CHECK-NEXT:   PseudoRET implicit $x10
 102   ; CHECK-NEXT: {{  $}}
 103   ; CHECK-NEXT: bb.8.default:
 104   ; CHECK-NEXT:   [[ADDI9:%[0-9]+]]:gpr = ADDI $x0, 1000
 105   ; CHECK-NEXT:   $x10 = COPY [[ADDI9]]
 106   ; CHECK-NEXT:   PseudoRET implicit $x10
 107   bb.1.entry:
 108     successors: %bb.8, %bb.9
 109     liveins: $x10
 110
 111     %1:gprb(s64) = COPY $x10
 112     %2:gprb(s64) = G_ASSERT_SEXT %1, 32
 113     %7:gprb(s64) = G_CONSTANT i64 5
 114     %3:gprb(s64) = G_SEXT_INREG %2, 32
 115     %4:gprb(s64) = G_CONSTANT i64 1
 116     %5:gprb(s64) = G_SUB %3, %4
 117     %26:gprb(s64) = G_ICMP intpred(ugt), %5(s64), %7
 118     G_BRCOND %26(s64), %bb.8
 119
 120   bb.9.entry:
 121     successors: %bb.2, %bb.3, %bb.4, %bb.5, %bb.6, %bb.7
 122
 123     %10:gprb(p0) = G_JUMP_TABLE %jump-table.0
 124     G_BRJT %10(p0), %jump-table.0, %5(s64)
 125
 126   bb.2.bb1:
 127     %22:gprb(s64) = G_CONSTANT i64 4
 128     $x10 = COPY %22(s64)
 129     PseudoRET implicit $x10
 130
 131   bb.3.bb2:
 132     %20:gprb(s64) = G_CONSTANT i64 3
 133     $x10 = COPY %20(s64)
 134     PseudoRET implicit $x10
 135
 136   bb.4.bb3:
 137     %18:gprb(s64) = G_CONSTANT i64 2
 138     $x10 = COPY %18(s64)
 139     PseudoRET implicit $x10
 140
 141   bb.5.bb4:
 142     %16:gprb(s64) = G_CONSTANT i64 1
 143     $x10 = COPY %16(s64)
 144     PseudoRET implicit $x10
 145
 146   bb.6.bb5:
 147     %14:gprb(s64) = G_CONSTANT i64 100
 148     $x10 = COPY %14(s64)
 149     PseudoRET implicit $x10
 150
 151   bb.7.bb6:
 152     %12:gprb(s64) = G_CONSTANT i64 200
 153     $x10 = COPY %12(s64)
 154     PseudoRET implicit $x10
 155
 156   bb.8.default:
 157     %24:gprb(s64) = G_CONSTANT i64 1000
 158     $x10 = COPY %24(s64)
 159     PseudoRET implicit $x10
 160
 161 ...