llvm/test/CodeGen/AArch64/sve-intrinsics-counting-elems.ll

   1 ; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sve < %s | FileCheck %s
   2
   3 ;
   4 ; CNTB
   5 ;
   6
   7 define i64 @cntb() {
   8 ; CHECK-LABEL: cntb:
   9 ; CHECK: cntb x0, vl2
  10 ; CHECK-NEXT: ret
  11   %out = call i64 @llvm.aarch64.sve.cntb(i32 2)
  12   ret i64 %out
  13 }
  14
  15 define i64 @cntb_mul3() {
  16 ; CHECK-LABEL: cntb_mul3:
  17 ; CHECK: cntb x0, vl6, mul #3
  18 ; CHECK-NEXT: ret
  19   %cnt = call i64 @llvm.aarch64.sve.cntb(i32 6)
  20   %out = mul i64 %cnt, 3
  21   ret i64 %out
  22 }
  23
  24 define i64 @cntb_mul4() {
  25 ; CHECK-LABEL: cntb_mul4:
  26 ; CHECK: cntb x0, vl8, mul #4
  27 ; CHECK-NEXT: ret
  28   %cnt = call i64 @llvm.aarch64.sve.cntb(i32 8)
  29   %out = mul i64 %cnt, 4
  30   ret i64 %out
  31 }
  32
  33 ;
  34 ; CNTH
  35 ;
  36
  37 define i64 @cnth() {
  38 ; CHECK-LABEL: cnth:
  39 ; CHECK: cnth x0, vl3
  40 ; CHECK-NEXT: ret
  41   %out = call i64 @llvm.aarch64.sve.cnth(i32 3)
  42   ret i64 %out
  43 }
  44
  45 define i64 @cnth_mul5() {
  46 ; CHECK-LABEL: cnth_mul5:
  47 ; CHECK: cnth x0, vl7, mul #5
  48 ; CHECK-NEXT: ret
  49   %cnt = call i64 @llvm.aarch64.sve.cnth(i32 7)
  50   %out = mul i64 %cnt, 5
  51   ret i64 %out
  52 }
  53
  54 define i64 @cnth_mul8() {
  55 ; CHECK-LABEL: cnth_mul8:
  56 ; CHECK: cnth x0, vl5, mul #8
  57 ; CHECK-NEXT: ret
  58   %cnt = call i64 @llvm.aarch64.sve.cnth(i32 5)
  59   %out = mul i64 %cnt, 8
  60   ret i64 %out
  61 }
  62
  63 ;
  64 ; CNTW
  65 ;
  66
  67 define i64 @cntw() {
  68 ; CHECK-LABEL: cntw:
  69 ; CHECK: cntw x0, vl4
  70 ; CHECK-NEXT: ret
  71   %out = call i64 @llvm.aarch64.sve.cntw(i32 4)
  72   ret i64 %out
  73 }
  74
  75 define i64 @cntw_mul11() {
  76 ; CHECK-LABEL: cntw_mul11:
  77 ; CHECK: cntw x0, vl8, mul #11
  78 ; CHECK-NEXT: ret
  79   %cnt = call i64 @llvm.aarch64.sve.cntw(i32 8)
  80   %out = mul i64 %cnt, 11
  81   ret i64 %out
  82 }
  83
  84 define i64 @cntw_mul2() {
  85 ; CHECK-LABEL: cntw_mul2:
  86 ; CHECK: cntw x0, vl6, mul #2
  87 ; CHECK-NEXT: ret
  88   %cnt = call i64 @llvm.aarch64.sve.cntw(i32 6)
  89   %out = mul i64 %cnt, 2
  90   ret i64 %out
  91 }
  92
  93 ;
  94 ; CNTD
  95 ;
  96
  97 define i64 @cntd() {
  98 ; CHECK-LABEL: cntd:
  99 ; CHECK: cntd x0, vl5
 100 ; CHECK-NEXT: ret
 101   %out = call i64 @llvm.aarch64.sve.cntd(i32 5)
 102   ret i64 %out
 103 }
 104
 105 define i64 @cntd_mul15() {
 106 ; CHECK-LABEL: cntd_mul15:
 107 ; CHECK: cntd x0, vl16, mul #15
 108 ; CHECK-NEXT: ret
 109   %cnt = call i64 @llvm.aarch64.sve.cntd(i32 9)
 110   %out = mul i64 %cnt, 15
 111   ret i64 %out
 112 }
 113
 114 define i64 @cntd_mul16() {
 115 ; CHECK-LABEL: cntd_mul16:
 116 ; CHECK: cntd x0, vl32, mul #16
 117 ; CHECK-NEXT: ret
 118   %cnt = call i64 @llvm.aarch64.sve.cntd(i32 10)
 119   %out = mul i64 %cnt, 16
 120   ret i64 %out
 121 }
 122
 123 ;
 124 ; CNTP
 125 ;
 126
 127 define i64 @cntp_b8(<vscale x 16 x i1> %pg, <vscale x 16 x i1> %a) {
 128 ; CHECK-LABEL: cntp_b8:
 129 ; CHECK: cntp x0, p0, p1.b
 130 ; CHECK-NEXT: ret
 131   %out = call i64 @llvm.aarch64.sve.cntp.nxv16i1(<vscale x 16 x i1> %pg,
 132                                                  <vscale x 16 x i1> %a)
 133   ret i64 %out
 134 }
 135
 136 define i64 @cntp_b16(<vscale x 8 x i1> %pg, <vscale x 8 x i1> %a) {
 137 ; CHECK-LABEL: cntp_b16:
 138 ; CHECK: cntp x0, p0, p1.h
 139 ; CHECK-NEXT: ret
 140   %out = call i64 @llvm.aarch64.sve.cntp.nxv8i1(<vscale x 8 x i1> %pg,
 141                                                 <vscale x 8 x i1> %a)
 142   ret i64 %out
 143 }
 144
 145 define i64 @cntp_b32(<vscale x 4 x i1> %pg, <vscale x 4 x i1> %a) {
 146 ; CHECK-LABEL: cntp_b32:
 147 ; CHECK: cntp x0, p0, p1.s
 148 ; CHECK-NEXT: ret
 149   %out = call i64 @llvm.aarch64.sve.cntp.nxv4i1(<vscale x 4 x i1> %pg,
 150                                                 <vscale x 4 x i1> %a)
 151   ret i64 %out
 152 }
 153
 154 define i64 @cntp_b64(<vscale x 2 x i1> %pg, <vscale x 2 x i1> %a) {
 155 ; CHECK-LABEL: cntp_b64:
 156 ; CHECK: cntp x0, p0, p1.d
 157 ; CHECK-NEXT: ret
 158   %out = call i64 @llvm.aarch64.sve.cntp.nxv2i1(<vscale x 2 x i1> %pg,
 159                                                 <vscale x 2 x i1> %a)
 160   ret i64 %out
 161 }
 162
 163 declare i64 @llvm.aarch64.sve.cntb(i32 %pattern)
 164 declare i64 @llvm.aarch64.sve.cnth(i32 %pattern)
 165 declare i64 @llvm.aarch64.sve.cntw(i32 %pattern)
 166 declare i64 @llvm.aarch64.sve.cntd(i32 %pattern)
 167
 168 declare i64 @llvm.aarch64.sve.cntp.nxv16i1(<vscale x 16 x i1>, <vscale x 16 x i1>)
 169 declare i64 @llvm.aarch64.sve.cntp.nxv8i1(<vscale x 8 x i1>, <vscale x 8 x i1>)
 170 declare i64 @llvm.aarch64.sve.cntp.nxv4i1(<vscale x 4 x i1>, <vscale x 4 x i1>)
 171 declare i64 @llvm.aarch64.sve.cntp.nxv2i1(<vscale x 2 x i1>, <vscale x 2 x i1>)