llvm/test/CodeGen/AArch64/sve-intrinsics-pred-operations.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sve < %s | FileCheck %s
   3 ; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sme -force-streaming < %s | FileCheck %s
   4
   5 ;
   6 ; BRKA
   7 ;
   8
   9 define <vscale x 16 x i1> @brka_m_b8(<vscale x 16 x i1> %inactive, <vscale x 16 x i1> %pg, <vscale x 16 x i1> %a) {
  10 ; CHECK-LABEL: brka_m_b8:
  11 ; CHECK:       // %bb.0:
  12 ; CHECK-NEXT:    brka p0.b, p1/m, p2.b
  13 ; CHECK-NEXT:    ret
  14   %out = call <vscale x 16 x i1> @llvm.aarch64.sve.brka.nxv16i1(<vscale x 16 x i1> %inactive,
  15                                                                 <vscale x 16 x i1> %pg,
  16                                                                 <vscale x 16 x i1> %a)
  17   ret <vscale x 16 x i1> %out
  18 }
  19
  20 define <vscale x 16 x i1> @brka_z_b8(<vscale x 16 x i1> %pg, <vscale x 16 x i1> %a) {
  21 ; CHECK-LABEL: brka_z_b8:
  22 ; CHECK:       // %bb.0:
  23 ; CHECK-NEXT:    brka p0.b, p0/z, p1.b
  24 ; CHECK-NEXT:    ret
  25   %out = call <vscale x 16 x i1> @llvm.aarch64.sve.brka.z.nxv16i1(<vscale x 16 x i1> %pg,
  26                                                                   <vscale x 16 x i1> %a)
  27   ret <vscale x 16 x i1> %out
  28 }
  29
  30 ;
  31 ; BRKB
  32 ;
  33
  34 define <vscale x 16 x i1> @brkb_m_b8(<vscale x 16 x i1> %inactive, <vscale x 16 x i1> %pg, <vscale x 16 x i1> %a) {
  35 ; CHECK-LABEL: brkb_m_b8:
  36 ; CHECK:       // %bb.0:
  37 ; CHECK-NEXT:    brkb p0.b, p1/m, p2.b
  38 ; CHECK-NEXT:    ret
  39   %out = call <vscale x 16 x i1> @llvm.aarch64.sve.brkb.nxv16i1(<vscale x 16 x i1> %inactive,
  40                                                                 <vscale x 16 x i1> %pg,
  41                                                                 <vscale x 16 x i1> %a)
  42   ret <vscale x 16 x i1> %out
  43 }
  44
  45 define <vscale x 16 x i1> @brkb_z_b8(<vscale x 16 x i1> %pg, <vscale x 16 x i1> %a) {
  46 ; CHECK-LABEL: brkb_z_b8:
  47 ; CHECK:       // %bb.0:
  48 ; CHECK-NEXT:    brkb p0.b, p0/z, p1.b
  49 ; CHECK-NEXT:    ret
  50   %out = call <vscale x 16 x i1> @llvm.aarch64.sve.brkb.z.nxv16i1(<vscale x 16 x i1> %pg,
  51                                                                   <vscale x 16 x i1> %a)
  52   ret <vscale x 16 x i1> %out
  53 }
  54
  55 ;
  56 ; BRKN
  57 ;
  58
  59 define <vscale x 16 x i1> @brkn_b8(<vscale x 16 x i1> %pg, <vscale x 16 x i1> %a, <vscale x 16 x i1> %b) {
  60 ; CHECK-LABEL: brkn_b8:
  61 ; CHECK:       // %bb.0:
  62 ; CHECK-NEXT:    brkn p2.b, p0/z, p1.b, p2.b
  63 ; CHECK-NEXT:    mov p0.b, p2.b
  64 ; CHECK-NEXT:    ret
  65   %out = call <vscale x 16 x i1> @llvm.aarch64.sve.brkn.z.nxv16i1(<vscale x 16 x i1> %pg,
  66                                                                   <vscale x 16 x i1> %a,
  67                                                                   <vscale x 16 x i1> %b)
  68   ret <vscale x 16 x i1> %out
  69 }
  70
  71 ;
  72 ; BRKPA
  73 ;
  74
  75 define <vscale x 16 x i1> @brkpa_b8(<vscale x 16 x i1> %pg, <vscale x 16 x i1> %a, <vscale x 16 x i1> %b) {
  76 ; CHECK-LABEL: brkpa_b8:
  77 ; CHECK:       // %bb.0:
  78 ; CHECK-NEXT:    brkpa p0.b, p0/z, p1.b, p2.b
  79 ; CHECK-NEXT:    ret
  80   %out = call <vscale x 16 x i1> @llvm.aarch64.sve.brkpa.z.nxv16i1(<vscale x 16 x i1> %pg,
  81                                                                    <vscale x 16 x i1> %a,
  82                                                                    <vscale x 16 x i1> %b)
  83   ret <vscale x 16 x i1> %out
  84 }
  85
  86 ;
  87 ; BRKPB
  88 ;
  89
  90 define <vscale x 16 x i1> @brkpb_b8(<vscale x 16 x i1> %pg, <vscale x 16 x i1> %a, <vscale x 16 x i1> %b) {
  91 ; CHECK-LABEL: brkpb_b8:
  92 ; CHECK:       // %bb.0:
  93 ; CHECK-NEXT:    brkpb p0.b, p0/z, p1.b, p2.b
  94 ; CHECK-NEXT:    ret
  95   %out = call <vscale x 16 x i1> @llvm.aarch64.sve.brkpb.z.nxv16i1(<vscale x 16 x i1> %pg,
  96                                                                    <vscale x 16 x i1> %a,
  97                                                                    <vscale x 16 x i1> %b)
  98   ret <vscale x 16 x i1> %out
  99 }
 100
 101 ;
 102 ; PFIRST
 103 ;
 104
 105 define <vscale x 16 x i1> @pfirst_b8(<vscale x 16 x i1> %pg, <vscale x 16 x i1> %a) {
 106 ; CHECK-LABEL: pfirst_b8:
 107 ; CHECK:       // %bb.0:
 108 ; CHECK-NEXT:    pfirst p1.b, p0, p1.b
 109 ; CHECK-NEXT:    mov p0.b, p1.b
 110 ; CHECK-NEXT:    ret
 111   %out = call <vscale x 16 x i1> @llvm.aarch64.sve.pfirst.nxv16i1(<vscale x 16 x i1> %pg,
 112                                                                   <vscale x 16 x i1> %a)
 113   ret <vscale x 16 x i1> %out
 114 }
 115
 116 ;
 117 ; PNEXT
 118 ;
 119
 120 define <vscale x 16 x i1> @pnext_b8(<vscale x 16 x i1> %pg, <vscale x 16 x i1> %a) {
 121 ; CHECK-LABEL: pnext_b8:
 122 ; CHECK:       // %bb.0:
 123 ; CHECK-NEXT:    pnext p1.b, p0, p1.b
 124 ; CHECK-NEXT:    mov p0.b, p1.b
 125 ; CHECK-NEXT:    ret
 126   %out = call <vscale x 16 x i1> @llvm.aarch64.sve.pnext.nxv16i1(<vscale x 16 x i1> %pg,
 127                                                                  <vscale x 16 x i1> %a)
 128   ret <vscale x 16 x i1> %out
 129 }
 130
 131 define <vscale x 8 x i1> @pnext_b16(<vscale x 8 x i1> %pg, <vscale x 8 x i1> %a) {
 132 ; CHECK-LABEL: pnext_b16:
 133 ; CHECK:       // %bb.0:
 134 ; CHECK-NEXT:    pnext p1.h, p0, p1.h
 135 ; CHECK-NEXT:    mov p0.b, p1.b
 136 ; CHECK-NEXT:    ret
 137   %out = call <vscale x 8 x i1> @llvm.aarch64.sve.pnext.nxv8i1(<vscale x 8 x i1> %pg,
 138                                                                <vscale x 8 x i1> %a)
 139   ret <vscale x 8 x i1> %out
 140 }
 141
 142 define <vscale x 4 x i1> @pnext_b32(<vscale x 4 x i1> %pg, <vscale x 4 x i1> %a) {
 143 ; CHECK-LABEL: pnext_b32:
 144 ; CHECK:       // %bb.0:
 145 ; CHECK-NEXT:    pnext p1.s, p0, p1.s
 146 ; CHECK-NEXT:    mov p0.b, p1.b
 147 ; CHECK-NEXT:    ret
 148   %out = call <vscale x 4 x i1> @llvm.aarch64.sve.pnext.nxv4i1(<vscale x 4 x i1> %pg,
 149                                                                <vscale x 4 x i1> %a)
 150   ret <vscale x 4 x i1> %out
 151 }
 152
 153 define <vscale x 2 x i1> @pnext_b64(<vscale x 2 x i1> %pg, <vscale x 2 x i1> %a) {
 154 ; CHECK-LABEL: pnext_b64:
 155 ; CHECK:       // %bb.0:
 156 ; CHECK-NEXT:    pnext p1.d, p0, p1.d
 157 ; CHECK-NEXT:    mov p0.b, p1.b
 158 ; CHECK-NEXT:    ret
 159   %out = call <vscale x 2 x i1> @llvm.aarch64.sve.pnext.nxv2i1(<vscale x 2 x i1> %pg,
 160                                                                <vscale x 2 x i1> %a)
 161   ret <vscale x 2 x i1> %out
 162 }
 163
 164 ;
 165 ; PUNPKHI
 166 ;
 167
 168 define <vscale x 8 x i1> @punpkhi_b16(<vscale x 16 x i1> %a) {
 169 ; CHECK-LABEL: punpkhi_b16:
 170 ; CHECK:       // %bb.0:
 171 ; CHECK-NEXT:    punpkhi p0.h, p0.b
 172 ; CHECK-NEXT:    ret
 173   %res = call <vscale x 8 x i1> @llvm.aarch64.sve.punpkhi.nxv8i1(<vscale x 16 x i1> %a)
 174   ret <vscale x 8 x i1> %res
 175 }
 176
 177 define <vscale x 4 x i1> @punpkhi_b8(<vscale x 8 x i1> %a) {
 178 ; CHECK-LABEL: punpkhi_b8:
 179 ; CHECK:       // %bb.0:
 180 ; CHECK-NEXT:    punpkhi p0.h, p0.b
 181 ; CHECK-NEXT:    ret
 182   %res = call <vscale x 4 x i1> @llvm.aarch64.sve.punpkhi.nxv4i1(<vscale x 8 x i1> %a)
 183   ret <vscale x 4 x i1> %res
 184 }
 185
 186 define <vscale x 2 x i1> @punpkhi_b4(<vscale x 4 x i1> %a) {
 187 ; CHECK-LABEL: punpkhi_b4:
 188 ; CHECK:       // %bb.0:
 189 ; CHECK-NEXT:    punpkhi p0.h, p0.b
 190 ; CHECK-NEXT:    ret
 191   %res = call <vscale x 2 x i1> @llvm.aarch64.sve.punpkhi.nxv2i1(<vscale x 4 x i1> %a)
 192   ret <vscale x 2 x i1> %res
 193 }
 194
 195 ;
 196 ; PUNPKLO
 197 ;
 198
 199 define <vscale x 8 x i1> @punpklo_b16(<vscale x 16 x i1> %a) {
 200 ; CHECK-LABEL: punpklo_b16:
 201 ; CHECK:       // %bb.0:
 202 ; CHECK-NEXT:    punpklo p0.h, p0.b
 203 ; CHECK-NEXT:    ret
 204   %res = call <vscale x 8 x i1> @llvm.aarch64.sve.punpklo.nxv8i1(<vscale x 16 x i1> %a)
 205   ret <vscale x 8 x i1> %res
 206 }
 207
 208 define <vscale x 4 x i1> @punpklo_b8(<vscale x 8 x i1> %a) {
 209 ; CHECK-LABEL: punpklo_b8:
 210 ; CHECK:       // %bb.0:
 211 ; CHECK-NEXT:    punpklo p0.h, p0.b
 212 ; CHECK-NEXT:    ret
 213   %res = call <vscale x 4 x i1> @llvm.aarch64.sve.punpklo.nxv4i1(<vscale x 8 x i1> %a)
 214   ret <vscale x 4 x i1> %res
 215 }
 216
 217 define <vscale x 2 x i1> @punpklo_b4(<vscale x 4 x i1> %a) {
 218 ; CHECK-LABEL: punpklo_b4:
 219 ; CHECK:       // %bb.0:
 220 ; CHECK-NEXT:    punpklo p0.h, p0.b
 221 ; CHECK-NEXT:    ret
 222   %res = call <vscale x 2 x i1> @llvm.aarch64.sve.punpklo.nxv2i1(<vscale x 4 x i1> %a)
 223   ret <vscale x 2 x i1> %res
 224 }
 225
 226 declare <vscale x 16 x i1> @llvm.aarch64.sve.brka.nxv16i1(<vscale x 16 x i1>, <vscale x 16 x i1>, <vscale x 16 x i1>)
 227 declare <vscale x 16 x i1> @llvm.aarch64.sve.brka.z.nxv16i1(<vscale x 16 x i1>, <vscale x 16 x i1>)
 228 declare <vscale x 16 x i1> @llvm.aarch64.sve.brkb.nxv16i1(<vscale x 16 x i1>, <vscale x 16 x i1>, <vscale x 16 x i1>)
 229 declare <vscale x 16 x i1> @llvm.aarch64.sve.brkb.z.nxv16i1(<vscale x 16 x i1>, <vscale x 16 x i1>)
 230 declare <vscale x 16 x i1> @llvm.aarch64.sve.brkn.z.nxv16i1(<vscale x 16 x i1>, <vscale x 16 x i1>, <vscale x 16 x i1>)
 231 declare <vscale x 16 x i1> @llvm.aarch64.sve.brkpa.z.nxv16i1(<vscale x 16 x i1>, <vscale x 16 x i1>, <vscale x 16 x i1>)
 232 declare <vscale x 16 x i1> @llvm.aarch64.sve.brkpb.z.nxv16i1(<vscale x 16 x i1>, <vscale x 16 x i1>, <vscale x 16 x i1>)
 233
 234 declare <vscale x 16 x i1> @llvm.aarch64.sve.pfirst.nxv16i1(<vscale x 16 x i1>, <vscale x 16 x i1>)
 235
 236 declare <vscale x 16 x i1> @llvm.aarch64.sve.pnext.nxv16i1(<vscale x 16 x i1>, <vscale x 16 x i1>)
 237 declare <vscale x 8 x i1> @llvm.aarch64.sve.pnext.nxv8i1(<vscale x 8 x i1>, <vscale x 8 x i1>)
 238 declare <vscale x 4 x i1> @llvm.aarch64.sve.pnext.nxv4i1(<vscale x 4 x i1>, <vscale x 4 x i1>)
 239 declare <vscale x 2 x i1> @llvm.aarch64.sve.pnext.nxv2i1(<vscale x 2 x i1>, <vscale x 2 x i1>)
 240
 241 declare <vscale x 8 x i1> @llvm.aarch64.sve.punpkhi.nxv8i1(<vscale x 16 x i1>)
 242 declare <vscale x 4 x i1> @llvm.aarch64.sve.punpkhi.nxv4i1(<vscale x 8 x i1>)
 243 declare <vscale x 2 x i1> @llvm.aarch64.sve.punpkhi.nxv2i1(<vscale x 4 x i1>)
 244
 245 declare <vscale x 8 x i1> @llvm.aarch64.sve.punpklo.nxv8i1(<vscale x 16 x i1>)
 246 declare <vscale x 4 x i1> @llvm.aarch64.sve.punpklo.nxv4i1(<vscale x 8 x i1>)
 247 declare <vscale x 2 x i1> @llvm.aarch64.sve.punpklo.nxv2i1(<vscale x 4 x i1>)