llvm/test/CodeGen/AArch64/GlobalISel/form-bitfield-extract-from-shr.mir

   1 # NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py
   2 # RUN: llc -mtriple aarch64 -run-pass=aarch64-postlegalizer-combiner -verify-machineinstrs %s -o - | FileCheck %s
   3
   4 # Check that we can fold a G_ASHR/G_LSHR fed by a G_SHL into a G_SBFX/G_UBFX.
   5
   6 ---
   7 name:            apply_ashr_shl_to_sbfx
   8 legalized: true
   9 body:             |
  10   bb.0.entry:
  11     ; CHECK-LABEL: name: apply_ashr_shl_to_sbfx
  12     ; CHECK: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
  13     ; CHECK: [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 16
  14     ; CHECK: [[C1:%[0-9]+]]:_(s32) = G_CONSTANT i32 11
  15     ; CHECK: [[SBFX:%[0-9]+]]:_(s32) = G_SBFX [[COPY]], [[C1]](s32), [[C]]
  16     ; CHECK: $w0 = COPY [[SBFX]](s32)
  17     %0:_(s32) = COPY $w0
  18     %1:_(s32) = G_CONSTANT i32 5
  19     %2:_(s32) = G_CONSTANT i32 16
  20     %3:_(s32) = G_SHL %0, %1
  21     %4:_(s32) = G_ASHR %3, %2
  22     $w0 = COPY %4(s32)
  23 ...
  24
  25 ---
  26 name:            apply_ashr_shl_to_sbfx_lower_bound
  27 legalized: true
  28 body:             |
  29   bb.0.entry:
  30     ; CHECK-LABEL: name: apply_ashr_shl_to_sbfx_lower_bound
  31     ; CHECK: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
  32     ; CHECK: [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 1
  33     ; CHECK: [[SBFX:%[0-9]+]]:_(s32) = G_SBFX [[COPY]], [[C]](s32), [[C]]
  34     ; CHECK: $w0 = COPY [[SBFX]](s32)
  35     %0:_(s32) = COPY $w0
  36     %1:_(s32) = G_CONSTANT i32 30
  37     %2:_(s32) = G_CONSTANT i32 31
  38     %3:_(s32) = G_SHL %0, %1
  39     %4:_(s32) = G_ASHR %3, %2
  40     $w0 = COPY %4(s32)
  41 ...
  42
  43 ---
  44 name:            apply_ashr_shl_to_sbfx_upper_bound
  45 legalized: true
  46 body:             |
  47   bb.0.entry:
  48     ; CHECK-LABEL: name: apply_ashr_shl_to_sbfx_upper_bound
  49     ; CHECK: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
  50     ; CHECK: [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 1
  51     ; CHECK: [[C1:%[0-9]+]]:_(s32) = G_CONSTANT i32 30
  52     ; CHECK: [[SBFX:%[0-9]+]]:_(s32) = G_SBFX [[COPY]], [[C1]](s32), [[C]]
  53     ; CHECK: $w0 = COPY [[SBFX]](s32)
  54     %0:_(s32) = COPY $w0
  55     %1:_(s32) = G_CONSTANT i32 1
  56     %2:_(s32) = G_CONSTANT i32 31
  57     %3:_(s32) = G_SHL %0, %1
  58     %4:_(s32) = G_ASHR %3, %2
  59     $w0 = COPY %4(s32)
  60 ...
  61
  62 ---
  63 name:            apply_lshr_shl_to_ubfx
  64 legalized: true
  65 body:             |
  66   bb.0.entry:
  67     ; CHECK-LABEL: name: apply_lshr_shl_to_ubfx
  68     ; CHECK: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
  69     ; CHECK: [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 16
  70     ; CHECK: [[C1:%[0-9]+]]:_(s32) = G_CONSTANT i32 11
  71     ; CHECK: [[UBFX:%[0-9]+]]:_(s32) = G_UBFX [[COPY]], [[C1]](s32), [[C]]
  72     ; CHECK: $w0 = COPY [[UBFX]](s32)
  73     %0:_(s32) = COPY $w0
  74     %1:_(s32) = G_CONSTANT i32 5
  75     %2:_(s32) = G_CONSTANT i32 16
  76     %3:_(s32) = G_SHL %0, %1
  77     %4:_(s32) = G_LSHR %3, %2
  78     $w0 = COPY %4(s32)
  79 ...
  80
  81 ---
  82 name:            apply_lshr_shl_to_ubfx_lower_bound
  83 legalized: true
  84 body:             |
  85   bb.0.entry:
  86     ; CHECK-LABEL: name: apply_lshr_shl_to_ubfx_lower_bound
  87     ; CHECK: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
  88     ; CHECK: [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 1
  89     ; CHECK: [[UBFX:%[0-9]+]]:_(s32) = G_UBFX [[COPY]], [[C]](s32), [[C]]
  90     ; CHECK: $w0 = COPY [[UBFX]](s32)
  91     %0:_(s32) = COPY $w0
  92     %1:_(s32) = G_CONSTANT i32 30
  93     %2:_(s32) = G_CONSTANT i32 31
  94     %3:_(s32) = G_SHL %0, %1
  95     %4:_(s32) = G_LSHR %3, %2
  96     $w0 = COPY %4(s32)
  97 ...
  98
  99 ---
 100 name:            apply_lshr_shl_to_ubfx_upper_bound
 101 legalized: true
 102 body:             |
 103   bb.0.entry:
 104     ; CHECK-LABEL: name: apply_lshr_shl_to_ubfx_upper_bound
 105     ; CHECK: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
 106     ; CHECK: [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 1
 107     ; CHECK: [[C1:%[0-9]+]]:_(s32) = G_CONSTANT i32 30
 108     ; CHECK: [[UBFX:%[0-9]+]]:_(s32) = G_UBFX [[COPY]], [[C1]](s32), [[C]]
 109     ; CHECK: $w0 = COPY [[UBFX]](s32)
 110     %0:_(s32) = COPY $w0
 111     %1:_(s32) = G_CONSTANT i32 1
 112     %2:_(s32) = G_CONSTANT i32 31
 113     %3:_(s32) = G_SHL %0, %1
 114     %4:_(s32) = G_LSHR %3, %2
 115     $w0 = COPY %4(s32)
 116 ...
 117
 118 ---
 119 name:            dont_apply_pos_out_of_bounds
 120 legalized: true
 121 body:             |
 122   bb.0.entry:
 123     ; CHECK-LABEL: name: dont_apply_pos_out_of_bounds
 124     ; CHECK: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
 125     ; CHECK: [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 5
 126     ; CHECK: [[C1:%[0-9]+]]:_(s32) = G_CONSTANT i32 2
 127     ; CHECK: [[SHL:%[0-9]+]]:_(s32) = G_SHL [[COPY]], [[C]](s32)
 128     ; CHECK: [[ASHR:%[0-9]+]]:_(s32) = G_ASHR [[SHL]], [[C1]](s32)
 129     ; CHECK: $w0 = COPY [[ASHR]](s32)
 130     %0:_(s32) = COPY $w0
 131     %1:_(s32) = G_CONSTANT i32 5
 132     %2:_(s32) = G_CONSTANT i32 2
 133     %3:_(s32) = G_SHL %0, %1
 134     %4:_(s32) = G_ASHR %3, %2
 135     $w0 = COPY %4(s32)
 136 ...
 137
 138 ---
 139 name:            dont_apply_no_constant
 140 legalized: true
 141 body:             |
 142   bb.0.entry:
 143     ; CHECK-LABEL: name: dont_apply_no_constant
 144     ; CHECK: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
 145     ; CHECK: [[COPY1:%[0-9]+]]:_(s32) = COPY $w1
 146     ; CHECK: [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 2
 147     ; CHECK: [[SHL:%[0-9]+]]:_(s32) = G_SHL [[COPY]], [[COPY1]](s32)
 148     ; CHECK: [[ASHR:%[0-9]+]]:_(s32) = G_ASHR [[SHL]], [[C]](s32)
 149     ; CHECK: $w0 = COPY [[ASHR]](s32)
 150     %0:_(s32) = COPY $w0
 151     %1:_(s32) = COPY $w1
 152     %2:_(s32) = G_CONSTANT i32 2
 153     %3:_(s32) = G_SHL %0, %1
 154     %4:_(s32) = G_ASHR %3, %2
 155     $w0 = COPY %4(s32)
 156 ...
 157
 158 ---
 159 name:            dont_apply_more_than_one_use
 160 legalized: true
 161 body:             |
 162   bb.0.entry:
 163     ; CHECK-LABEL: name: dont_apply_more_than_one_use
 164     ; CHECK: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
 165     ; CHECK: [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 5
 166     ; CHECK: [[C1:%[0-9]+]]:_(s32) = G_CONSTANT i32 16
 167     ; CHECK: [[SHL:%[0-9]+]]:_(s32) = G_SHL [[COPY]], [[C]](s32)
 168     ; CHECK: [[ASHR:%[0-9]+]]:_(s32) = G_ASHR [[SHL]], [[C1]](s32)
 169     ; CHECK: [[MUL:%[0-9]+]]:_(s32) = G_MUL [[SHL]], [[ASHR]]
 170     ; CHECK: $w0 = COPY [[MUL]](s32)
 171     %0:_(s32) = COPY $w0
 172     %1:_(s32) = G_CONSTANT i32 5
 173     %2:_(s32) = G_CONSTANT i32 16
 174     %3:_(s32) = G_SHL %0, %1
 175     %4:_(s32) = G_ASHR %3, %2
 176     %5:_(s32) = G_MUL %3, %4
 177     $w0 = COPY %5(s32)
 178 ...