llvm/test/Transforms/InstCombine/AArch64/sve-intrinsic-opts-counting-elems.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt -S -instcombine < %s | FileCheck %s
   3
   4 target triple = "aarch64-unknown-linux-gnu"
   5
   6 ;
   7 ; CNTB
   8 ;
   9
  10 define i64 @cntb_vl1() {
  11 ; CHECK-LABEL: @cntb_vl1(
  12 ; CHECK-NEXT:    ret i64 1
  13 ;
  14   %out = call i64 @llvm.aarch64.sve.cntb(i32 1)
  15   ret i64 %out
  16 }
  17
  18 define i64 @cntb_vl2() {
  19 ; CHECK-LABEL: @cntb_vl2(
  20 ; CHECK-NEXT:    ret i64 2
  21 ;
  22   %out = call i64 @llvm.aarch64.sve.cntb(i32 2)
  23   ret i64 %out
  24 }
  25
  26 define i64 @cntb_vl4() {
  27 ; CHECK-LABEL: @cntb_vl4(
  28 ; CHECK-NEXT:    ret i64 4
  29 ;
  30   %out = call i64 @llvm.aarch64.sve.cntb(i32 4)
  31   ret i64 %out
  32 }
  33
  34 define i64 @cntb_mul3() {
  35 ; CHECK-LABEL: @cntb_mul3(
  36 ; CHECK-NEXT:    ret i64 24
  37 ;
  38   %cnt = call i64 @llvm.aarch64.sve.cntb(i32 8)
  39   %out = mul i64 %cnt, 3
  40   ret i64 %out
  41 }
  42
  43 define i64 @cntb_mul4() {
  44 ; CHECK-LABEL: @cntb_mul4(
  45 ; CHECK-NEXT:    ret i64 64
  46 ;
  47   %cnt = call i64 @llvm.aarch64.sve.cntb(i32 9)
  48   %out = mul i64 %cnt, 4
  49   ret i64 %out
  50 }
  51
  52 define i64 @cntb_all() {
  53 ; CHECK-LABEL: @cntb_all(
  54 ; CHECK-NEXT:    [[TMP1:%.*]] = call i64 @llvm.vscale.i64()
  55 ; CHECK-NEXT:    [[OUT:%.*]] = shl i64 [[TMP1]], 4
  56 ; CHECK-NEXT:    ret i64 [[OUT]]
  57 ;
  58   %out = call i64 @llvm.aarch64.sve.cntb(i32 31)
  59   ret i64 %out
  60 }
  61
  62 ;
  63 ; CNTH
  64 ;
  65
  66 define i64 @cnth_vl1() {
  67 ; CHECK-LABEL: @cnth_vl1(
  68 ; CHECK-NEXT:    ret i64 1
  69 ;
  70   %out = call i64 @llvm.aarch64.sve.cnth(i32 1)
  71   ret i64 %out
  72 }
  73
  74 define i64 @cnth_vl2() {
  75 ; CHECK-LABEL: @cnth_vl2(
  76 ; CHECK-NEXT:    ret i64 2
  77 ;
  78   %out = call i64 @llvm.aarch64.sve.cnth(i32 2)
  79   ret i64 %out
  80 }
  81
  82 define i64 @cnth_vl4() {
  83 ; CHECK-LABEL: @cnth_vl4(
  84 ; CHECK-NEXT:    ret i64 4
  85 ;
  86   %out = call i64 @llvm.aarch64.sve.cnth(i32 4)
  87   ret i64 %out
  88 }
  89
  90 define i64 @cnth_mul3() {
  91 ; CHECK-LABEL: @cnth_mul3(
  92 ; CHECK-NEXT:    ret i64 24
  93 ;
  94   %cnt = call i64 @llvm.aarch64.sve.cnth(i32 8)
  95   %out = mul i64 %cnt, 3
  96   ret i64 %out
  97 }
  98
  99 define i64 @cnth_mul4() {
 100 ; CHECK-LABEL: @cnth_mul4(
 101 ; CHECK-NEXT:    [[CNT:%.*]] = call i64 @llvm.aarch64.sve.cnth(i32 9)
 102 ; CHECK-NEXT:    [[OUT:%.*]] = shl i64 [[CNT]], 2
 103 ; CHECK-NEXT:    ret i64 [[OUT]]
 104 ;
 105   %cnt = call i64 @llvm.aarch64.sve.cnth(i32 9)
 106   %out = mul i64 %cnt, 4
 107   ret i64 %out
 108 }
 109
 110 define i64 @cnth_all() {
 111 ; CHECK-LABEL: @cnth_all(
 112 ; CHECK-NEXT:    [[TMP1:%.*]] = call i64 @llvm.vscale.i64()
 113 ; CHECK-NEXT:    [[OUT:%.*]] = shl i64 [[TMP1]], 3
 114 ; CHECK-NEXT:    ret i64 [[OUT]]
 115 ;
 116   %out = call i64 @llvm.aarch64.sve.cnth(i32 31)
 117   ret i64 %out
 118 }
 119
 120 ;
 121 ; CNTW
 122 ;
 123
 124 define i64 @cntw_vl1() {
 125 ; CHECK-LABEL: @cntw_vl1(
 126 ; CHECK-NEXT:    ret i64 1
 127 ;
 128   %out = call i64 @llvm.aarch64.sve.cntw(i32 1)
 129   ret i64 %out
 130 }
 131
 132 define i64 @cntw_vl2() {
 133 ; CHECK-LABEL: @cntw_vl2(
 134 ; CHECK-NEXT:    ret i64 2
 135 ;
 136   %out = call i64 @llvm.aarch64.sve.cntw(i32 2)
 137   ret i64 %out
 138 }
 139
 140 define i64 @cntw_vl4() {
 141 ; CHECK-LABEL: @cntw_vl4(
 142 ; CHECK-NEXT:    ret i64 4
 143 ;
 144   %out = call i64 @llvm.aarch64.sve.cntw(i32 4)
 145   ret i64 %out
 146 }
 147
 148 define i64 @cntw_mul3() {
 149 ; CHECK-LABEL: @cntw_mul3(
 150 ; CHECK-NEXT:    [[CNT:%.*]] = call i64 @llvm.aarch64.sve.cntw(i32 8)
 151 ; CHECK-NEXT:    [[OUT:%.*]] = mul i64 [[CNT]], 3
 152 ; CHECK-NEXT:    ret i64 [[OUT]]
 153 ;
 154   %cnt = call i64 @llvm.aarch64.sve.cntw(i32 8)
 155   %out = mul i64 %cnt, 3
 156   ret i64 %out
 157 }
 158
 159 define i64 @cntw_mul4() {
 160 ; CHECK-LABEL: @cntw_mul4(
 161 ; CHECK-NEXT:    [[CNT:%.*]] = call i64 @llvm.aarch64.sve.cntw(i32 9)
 162 ; CHECK-NEXT:    [[OUT:%.*]] = shl i64 [[CNT]], 2
 163 ; CHECK-NEXT:    ret i64 [[OUT]]
 164 ;
 165   %cnt = call i64 @llvm.aarch64.sve.cntw(i32 9)
 166   %out = mul i64 %cnt, 4
 167   ret i64 %out
 168 }
 169
 170 define i64 @cntw_all() {
 171 ; CHECK-LABEL: @cntw_all(
 172 ; CHECK-NEXT:    [[TMP1:%.*]] = call i64 @llvm.vscale.i64()
 173 ; CHECK-NEXT:    [[OUT:%.*]] = shl i64 [[TMP1]], 2
 174 ; CHECK-NEXT:    ret i64 [[OUT]]
 175 ;
 176   %out = call i64 @llvm.aarch64.sve.cntw(i32 31)
 177   ret i64 %out
 178 }
 179
 180
 181 ;
 182 ; CNTD
 183 ;
 184
 185 define i64 @cntd_vl1() {
 186 ; CHECK-LABEL: @cntd_vl1(
 187 ; CHECK-NEXT:    ret i64 1
 188 ;
 189   %out = call i64 @llvm.aarch64.sve.cntd(i32 1)
 190   ret i64 %out
 191 }
 192
 193 define i64 @cntd_vl2() {
 194 ; CHECK-LABEL: @cntd_vl2(
 195 ; CHECK-NEXT:    ret i64 2
 196 ;
 197   %out = call i64 @llvm.aarch64.sve.cntd(i32 2)
 198   ret i64 %out
 199 }
 200
 201 define i64 @cntd_vl4() {
 202 ; CHECK-LABEL: @cntd_vl4(
 203 ; CHECK-NEXT:    [[OUT:%.*]] = call i64 @llvm.aarch64.sve.cntd(i32 4)
 204 ; CHECK-NEXT:    ret i64 [[OUT]]
 205 ;
 206   %out = call i64 @llvm.aarch64.sve.cntd(i32 4)
 207   ret i64 %out
 208 }
 209
 210 define i64 @cntd_mul3() {
 211 ; CHECK-LABEL: @cntd_mul3(
 212 ; CHECK-NEXT:    [[CNT:%.*]] = call i64 @llvm.aarch64.sve.cntd(i32 8)
 213 ; CHECK-NEXT:    [[OUT:%.*]] = mul i64 [[CNT]], 3
 214 ; CHECK-NEXT:    ret i64 [[OUT]]
 215 ;
 216   %cnt = call i64 @llvm.aarch64.sve.cntd(i32 8)
 217   %out = mul i64 %cnt, 3
 218   ret i64 %out
 219 }
 220
 221 define i64 @cntd_mul4() {
 222 ; CHECK-LABEL: @cntd_mul4(
 223 ; CHECK-NEXT:    [[CNT:%.*]] = call i64 @llvm.aarch64.sve.cntd(i32 9)
 224 ; CHECK-NEXT:    [[OUT:%.*]] = shl i64 [[CNT]], 2
 225 ; CHECK-NEXT:    ret i64 [[OUT]]
 226 ;
 227   %cnt = call i64 @llvm.aarch64.sve.cntd(i32 9)
 228   %out = mul i64 %cnt, 4
 229   ret i64 %out
 230 }
 231
 232 define i64 @cntd_all() {
 233 ; CHECK-LABEL: @cntd_all(
 234 ; CHECK-NEXT:    [[TMP1:%.*]] = call i64 @llvm.vscale.i64()
 235 ; CHECK-NEXT:    [[OUT:%.*]] = shl i64 [[TMP1]], 1
 236 ; CHECK-NEXT:    ret i64 [[OUT]]
 237 ;
 238   %out = call i64 @llvm.aarch64.sve.cntd(i32 31)
 239   ret i64 %out
 240 }
 241
 242
 243 declare i64 @llvm.aarch64.sve.cntb(i32 %pattern)
 244 declare i64 @llvm.aarch64.sve.cnth(i32 %pattern)
 245 declare i64 @llvm.aarch64.sve.cntw(i32 %pattern)
 246 declare i64 @llvm.aarch64.sve.cntd(i32 %pattern)
 247