test/Other/scalable-vectors-core-ir.ll

   1 ; RUN: opt -S -verify < %s | FileCheck %s
   2 target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128"
   3 target triple = "aarch64--linux-gnu"
   4
   5 ;; Check supported instructions are accepted without dropping 'vscale'.
   6 ;; Same order as the LangRef
   7
   8 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
   9 ;; Unary Operations
  10 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
  11
  12
  13 define <vscale x 2 x double> @fneg(<vscale x 2 x double> %val) {
  14 ; CHECK-LABEL: @fneg
  15 ; CHECK: %r = fneg <vscale x 2 x double> %val
  16 ; CHECK-NEXT: ret <vscale x 2 x double> %r
  17   %r = fneg <vscale x 2 x double> %val
  18   ret <vscale x 2 x double> %r
  19 }
  20
  21 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
  22 ;; Binary Operations
  23 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
  24
  25 define <vscale x 8 x i16> @add(<vscale x 8 x i16> %a, <vscale x 8 x i16> %b) {
  26 ; CHECK-LABEL: @add
  27 ; CHECK: %r = add <vscale x 8 x i16> %a, %b
  28 ; CHECK-NEXT: ret <vscale x 8 x i16> %r
  29   %r = add <vscale x 8 x i16> %a, %b
  30   ret <vscale x 8 x i16> %r
  31 }
  32
  33 define <vscale x 4 x float> @fadd(<vscale x 4 x float> %a, <vscale x 4 x float> %b) {
  34 ; CHECK-LABEL: @fadd
  35 ; CHECK: %r = fadd <vscale x 4 x float> %a, %b
  36 ; CHECK-NEXT: ret <vscale x 4 x float> %r
  37   %r = fadd <vscale x 4 x float> %a, %b
  38   ret <vscale x 4 x float> %r
  39 }
  40
  41 define <vscale x 4 x i32> @sub(<vscale x 4 x i32> %a, <vscale x 4 x i32> %b) {
  42 ; CHECK-LABEL: @sub
  43 ; CHECK: %r = sub <vscale x 4 x i32> %a, %b
  44 ; CHECK-NEXT: ret <vscale x 4 x i32> %r
  45   %r = sub <vscale x 4 x i32> %a, %b
  46   ret <vscale x 4 x i32> %r
  47 }
  48
  49 define <vscale x 4 x float> @fsub(<vscale x 4 x float> %a, <vscale x 4 x float> %b) {
  50 ; CHECK-LABEL: @fsub
  51 ; CHECK: %r = fsub <vscale x 4 x float> %a, %b
  52 ; CHECK-NEXT: ret <vscale x 4 x float> %r
  53   %r = fsub <vscale x 4 x float> %a, %b
  54   ret <vscale x 4 x float> %r
  55 }
  56
  57 define <vscale x 4 x i32> @mul(<vscale x 4 x i32> %a, <vscale x 4 x i32> %b) {
  58 ; CHECK-LABEL: @mul
  59 ; CHECK: %r = mul <vscale x 4 x i32> %a, %b
  60 ; CHECK-NEXT: ret <vscale x 4 x i32> %r
  61   %r = mul <vscale x 4 x i32> %a, %b
  62   ret <vscale x 4 x i32> %r
  63 }
  64
  65 define <vscale x 4 x float> @fmul(<vscale x 4 x float> %a, <vscale x 4 x float> %b) {
  66 ; CHECK-LABEL: @fmul
  67 ; CHECK: %r = fmul <vscale x 4 x float> %a, %b
  68 ; CHECK-NEXT: ret <vscale x 4 x float> %r
  69   %r = fmul <vscale x 4 x float> %a, %b
  70   ret <vscale x 4 x float> %r
  71 }
  72
  73 define <vscale x 4 x i32> @udiv(<vscale x 4 x i32> %a, <vscale x 4 x i32> %b) {
  74 ; CHECK-LABEL: @udiv
  75 ; CHECK: %r = udiv <vscale x 4 x i32> %a, %b
  76 ; CHECK-NEXT: ret <vscale x 4 x i32> %r
  77   %r = udiv <vscale x 4 x i32> %a, %b
  78   ret <vscale x 4 x i32> %r
  79 }
  80
  81 define <vscale x 4 x i32> @sdiv(<vscale x 4 x i32> %a, <vscale x 4 x i32> %b) {
  82 ; CHECK-LABEL: @sdiv
  83 ; CHECK: %r = sdiv <vscale x 4 x i32> %a, %b
  84 ; CHECK-NEXT: ret <vscale x 4 x i32> %r
  85   %r = sdiv <vscale x 4 x i32> %a, %b
  86   ret <vscale x 4 x i32> %r
  87 }
  88
  89 define <vscale x 4 x float> @fdiv(<vscale x 4 x float> %a, <vscale x 4 x float> %b) {
  90 ; CHECK-LABEL: @fdiv
  91 ; CHECK: %r = fdiv <vscale x 4 x float> %a, %b
  92 ; CHECK-NEXT: ret <vscale x 4 x float> %r
  93   %r = fdiv <vscale x 4 x float> %a, %b
  94   ret <vscale x 4 x float> %r
  95 }
  96
  97 define <vscale x 4 x i32> @urem(<vscale x 4 x i32> %a, <vscale x 4 x i32> %b) {
  98 ; CHECK-LABEL: @urem
  99 ; CHECK: %r = urem <vscale x 4 x i32> %a, %b
 100 ; CHECK-NEXT: ret <vscale x 4 x i32> %r
 101   %r = urem <vscale x 4 x i32> %a, %b
 102   ret <vscale x 4 x i32> %r
 103 }
 104
 105 define <vscale x 4 x i32> @srem(<vscale x 4 x i32> %a, <vscale x 4 x i32> %b) {
 106 ; CHECK-LABEL: @srem
 107 ; CHECK: %r = srem <vscale x 4 x i32> %a, %b
 108 ; CHECK-NEXT: ret <vscale x 4 x i32> %r
 109   %r = srem <vscale x 4 x i32> %a, %b
 110   ret <vscale x 4 x i32> %r
 111 }
 112
 113 define <vscale x 4 x float> @frem(<vscale x 4 x float> %a, <vscale x 4 x float> %b) {
 114 ; CHECK-LABEL: @frem
 115 ; CHECK: %r = frem <vscale x 4 x float> %a, %b
 116 ; CHECK-NEXT: ret <vscale x 4 x float> %r
 117   %r = frem <vscale x 4 x float> %a, %b
 118   ret <vscale x 4 x float> %r
 119 }
 120
 121 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 122 ;; Bitwise Binary Operations
 123 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 124
 125 define <vscale x 4 x i32> @shl(<vscale x 4 x i32> %a, <vscale x 4 x i32> %b) {
 126 ; CHECK-LABEL: @shl
 127 ; CHECK: %r = shl <vscale x 4 x i32> %a, %b
 128 ; CHECK-NEXT: ret <vscale x 4 x i32> %r
 129   %r = shl <vscale x 4 x i32> %a, %b
 130   ret <vscale x 4 x i32> %r
 131 }
 132
 133 define <vscale x 4 x i32> @lshr(<vscale x 4 x i32> %a, <vscale x 4 x i32> %b) {
 134 ; CHECK-LABEL: @lshr
 135 ; CHECK: %r = lshr <vscale x 4 x i32> %a, %b
 136 ; CHECK-NEXT: ret <vscale x 4 x i32> %r
 137   %r = lshr <vscale x 4 x i32> %a, %b
 138   ret <vscale x 4 x i32> %r
 139 }
 140
 141 define <vscale x 4 x i32> @ashr(<vscale x 4 x i32> %a, <vscale x 4 x i32> %b) {
 142 ; CHECK-LABEL: @ashr
 143 ; CHECK: %r = ashr <vscale x 4 x i32> %a, %b
 144 ; CHECK-NEXT: ret <vscale x 4 x i32> %r
 145   %r = ashr <vscale x 4 x i32> %a, %b
 146   ret <vscale x 4 x i32> %r
 147 }
 148
 149 define <vscale x 4 x i32> @and(<vscale x 4 x i32> %a, <vscale x 4 x i32> %b) {
 150 ; CHECK-LABEL: @and
 151 ; CHECK: %r = and <vscale x 4 x i32> %a, %b
 152 ; CHECK-NEXT: ret <vscale x 4 x i32> %r
 153   %r = and <vscale x 4 x i32> %a, %b
 154   ret <vscale x 4 x i32> %r
 155 }
 156
 157 define <vscale x 4 x i32> @or(<vscale x 4 x i32> %a, <vscale x 4 x i32> %b) {
 158 ; CHECK-LABEL: @or
 159 ; CHECK: %r = or <vscale x 4 x i32> %a, %b
 160 ; CHECK-NEXT: ret <vscale x 4 x i32> %r
 161   %r = or <vscale x 4 x i32> %a, %b
 162   ret <vscale x 4 x i32> %r
 163 }
 164
 165 define <vscale x 4 x i32> @xor(<vscale x 4 x i32> %a, <vscale x 4 x i32> %b) {
 166 ; CHECK-LABEL: @xor
 167 ; CHECK: %r = xor <vscale x 4 x i32> %a, %b
 168 ; CHECK-NEXT: ret <vscale x 4 x i32> %r
 169   %r = xor <vscale x 4 x i32> %a, %b
 170   ret <vscale x 4 x i32> %r
 171 }
 172
 173 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 174 ;; Vector Operations
 175 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 176
 177 define i64 @extractelement(<vscale x 2 x i64> %val) {
 178 ; CHECK-LABEL: @extractelement
 179 ; CHECK: %r = extractelement <vscale x 2 x i64> %val, i32 0
 180 ; CHECK-NEXT: ret i64 %r
 181   %r = extractelement <vscale x 2 x i64> %val, i32 0
 182   ret i64 %r
 183 }
 184
 185 define <vscale x 16 x i8> @insertelement(<vscale x 16 x i8> %vec, i8 %ins) {
 186 ; CHECK-LABEL: @insertelement
 187 ; CHECK: %r = insertelement <vscale x 16 x i8> %vec, i8 %ins, i32 0
 188 ; CHECK-NEXT: ret <vscale x 16 x i8> %r
 189   %r = insertelement <vscale x 16 x i8> %vec, i8 %ins, i32 0
 190   ret <vscale x 16 x i8> %r
 191 }
 192
 193 define <vscale x 8 x half> @shufflevector(half %val) {
 194 ; CHECK-LABEL: @shufflevector
 195 ; CHECK: %insvec = insertelement <vscale x 8 x half> undef, half %val, i32 0
 196 ; CHECK-NEXT: %r = shufflevector <vscale x 8 x half> %insvec, <vscale x 8 x half> undef, <vscale x 8 x i32> zeroinitializer
 197 ; CHECK-NEXT: ret <vscale x 8 x half> %r
 198   %insvec = insertelement <vscale x 8 x half> undef, half %val, i32 0
 199   %r = shufflevector <vscale x 8 x half> %insvec, <vscale x 8 x half> undef, <vscale x 8 x i32> zeroinitializer
 200   ret <vscale x 8 x half> %r
 201 }
 202
 203 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 204 ;; Memory Access and Addressing Operations
 205 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 206
 207 define void @alloca() {
 208 ; CHECK-LABEL: @alloca
 209 ; CHECK: %vec = alloca <vscale x 4 x i32>
 210 ; CHECK-NEXT: ret void
 211   %vec = alloca <vscale x 4 x i32>
 212   ret void
 213 }
 214
 215 define <vscale x 2 x double> @load(<vscale x 2 x double>* %ptr) {
 216 ; CHECK-LABEL: @load
 217 ; CHECK: %r = load <vscale x 2 x double>, <vscale x 2 x double>* %ptr
 218 ; CHECK-NEXT: ret <vscale x 2 x double> %r
 219   %r = load <vscale x 2 x double>, <vscale x 2 x double>* %ptr
 220   ret <vscale x 2 x double> %r
 221 }
 222
 223 define void @store(<vscale x 4 x i32> %data, <vscale x 4 x i32>* %ptr) {
 224 ; CHECK-LABEL: @store
 225 ; CHECK: store <vscale x 4 x i32> %data, <vscale x 4 x i32>* %ptr
 226 ; CHECK-NEXT: ret void
 227   store <vscale x 4 x i32> %data, <vscale x 4 x i32>* %ptr
 228   ret void
 229 }
 230
 231 define <vscale x 4 x float>* @getelementptr(<vscale x 4 x float>* %base) {
 232 ; CHECK-LABEL: @getelementptr
 233 ; CHECK: %r = getelementptr <vscale x 4 x float>, <vscale x 4 x float>* %base, i64 0
 234 ; CHECK-NEXT: ret <vscale x 4 x float>* %r
 235   %r = getelementptr <vscale x 4 x float>, <vscale x 4 x float>* %base, i64 0
 236   ret <vscale x 4 x float>* %r
 237 }
 238
 239 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 240 ;; Conversion Operations
 241 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 242
 243 define <vscale x 4 x i32> @truncto(<vscale x 4 x i64> %val) {
 244 ; CHECK-LABEL: @truncto
 245 ; CHECK: %r = trunc <vscale x 4 x i64> %val to <vscale x 4 x i32>
 246 ; CHECK-NEXT: ret <vscale x 4 x i32> %r
 247   %r = trunc <vscale x 4 x i64> %val to <vscale x 4 x i32>
 248   ret <vscale x 4 x i32> %r
 249 }
 250
 251 define <vscale x 2 x i64> @zextto(<vscale x 2 x i16> %val) {
 252 ; CHECK-LABEL: @zextto
 253 ; CHECK: %r = zext <vscale x 2 x i16> %val to <vscale x 2 x i64>
 254 ; CHECK-NEXT: ret <vscale x 2 x i64> %r
 255   %r = zext <vscale x 2 x i16> %val to <vscale x 2 x i64>
 256   ret <vscale x 2 x i64> %r
 257 }
 258
 259 define <vscale x 4 x i32> @sextto(<vscale x 4 x i8> %val) {
 260 ; CHECK-LABEL: @sextto
 261 ; CHECK: %r = sext <vscale x 4 x i8> %val to <vscale x 4 x i32>
 262 ; CHECK-NEXT: ret <vscale x 4 x i32> %r
 263   %r = sext <vscale x 4 x i8> %val to <vscale x 4 x i32>
 264   ret <vscale x 4 x i32> %r
 265 }
 266
 267 define <vscale x 4 x half> @fptruncto(<vscale x 4 x float> %val) {
 268 ; CHECK-LABEL: @fptruncto
 269 ; CHECK: %r = fptrunc <vscale x 4 x float> %val to <vscale x 4 x half>
 270 ; CHECK-NEXT: ret <vscale x 4 x half> %r
 271   %r = fptrunc <vscale x 4 x float> %val to <vscale x 4 x half>
 272   ret <vscale x 4 x half> %r
 273 }
 274
 275 define <vscale x 2 x double> @fpextto(<vscale x 2 x half> %val) {
 276 ; CHECK-LABEL: @fpextto
 277 ; CHECK: %r = fpext <vscale x 2 x half> %val to <vscale x 2 x double>
 278 ; CHECK-NEXT: ret <vscale x 2 x double> %r
 279   %r = fpext <vscale x 2 x half> %val to <vscale x 2 x double>
 280   ret <vscale x 2 x double> %r
 281 }
 282
 283 define <vscale x 4 x i32> @fptouito(<vscale x 4 x float> %val) {
 284 ; CHECK-LABEL: @fptoui
 285 ; CHECK: %r = fptoui <vscale x 4 x float> %val to <vscale x 4 x i32>
 286 ; CHECK-NEXT: ret <vscale x 4 x i32> %r
 287   %r = fptoui <vscale x 4 x float> %val to <vscale x 4 x i32>
 288   ret <vscale x 4 x i32> %r
 289 }
 290
 291 define <vscale x 4 x i32> @fptosito(<vscale x 4 x float> %val) {
 292 ; CHECK-LABEL: @fptosi
 293 ; CHECK: %r = fptosi <vscale x 4 x float> %val to <vscale x 4 x i32>
 294 ; CHECK-NEXT: ret <vscale x 4 x i32> %r
 295   %r = fptosi <vscale x 4 x float> %val to <vscale x 4 x i32>
 296   ret <vscale x 4 x i32> %r
 297 }
 298
 299 define <vscale x 4 x float> @uitofpto(<vscale x 4 x i32> %val) {
 300 ; CHECK-LABEL: @uitofp
 301 ; CHECK: %r = uitofp <vscale x 4 x i32> %val to <vscale x 4 x float>
 302 ; CHECK-NEXT: ret <vscale x 4 x float> %r
 303   %r = uitofp <vscale x 4 x i32> %val to <vscale x 4 x float>
 304   ret <vscale x 4 x float> %r
 305 }
 306
 307 define <vscale x 4 x float> @sitofpto(<vscale x 4 x i32> %val) {
 308 ; CHECK-LABEL: @sitofp
 309 ; CHECK: %r = sitofp <vscale x 4 x i32> %val to <vscale x 4 x float>
 310 ; CHECK-NEXT: ret <vscale x 4 x float> %r
 311   %r = sitofp <vscale x 4 x i32> %val to <vscale x 4 x float>
 312   ret <vscale x 4 x float> %r
 313 }
 314
 315 define <vscale x 2 x i64> @ptrtointto(<vscale x 2 x i32*> %val) {
 316 ; CHECK-LABEL: @ptrtointto
 317 ; CHECK: %r = ptrtoint <vscale x 2 x i32*> %val to <vscale x 2 x i64>
 318 ; CHECK-NEXT: ret <vscale x 2 x i64> %r
 319   %r = ptrtoint <vscale x 2 x i32*> %val to <vscale x 2 x i64>
 320   ret <vscale x 2 x i64> %r
 321 }
 322
 323 define <vscale x 2 x i32*> @inttoptrto(<vscale x 2 x i64> %val) {
 324 ; CHECK-LABEL: @inttoptrto
 325 ; CHECK: %r = inttoptr <vscale x 2 x i64> %val to <vscale x 2 x i32*>
 326 ; CHECK-NEXT: ret <vscale x 2 x i32*> %r
 327   %r = inttoptr <vscale x 2 x i64> %val to <vscale x 2 x i32*>
 328   ret <vscale x 2 x i32*> %r
 329 }
 330
 331 define <vscale x 2 x i64> @bitcastto(<vscale x 2 x double> %a) {
 332 ; CHECK-LABEL: @bitcast
 333 ; CHECK: %r = bitcast <vscale x 2 x double> %a to <vscale x 2 x i64>
 334 ; CHECK-NEXT: ret <vscale x 2 x i64> %r
 335   %r = bitcast <vscale x 2 x double> %a to <vscale x 2 x i64>
 336   ret <vscale x 2 x i64> %r
 337 }
 338
 339 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 340 ;; Other Operations
 341 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 342
 343 define <vscale x 4 x i1> @icmp(<vscale x 4 x i32> %a, <vscale x 4 x i32> %b) {
 344 ; CHECK-LABEL: @icmp
 345 ; CHECK: %r = icmp eq <vscale x 4 x i32> %a, %b
 346 ; CHECK-NEXT: ret <vscale x 4 x i1> %r
 347   %r = icmp eq <vscale x 4 x i32> %a, %b
 348   ret <vscale x 4 x i1> %r
 349 }
 350
 351 define <vscale x 4 x i1> @fcmp(<vscale x 4 x float> %a, <vscale x 4 x float> %b) {
 352 ; CHECK-LABEL: @fcmp
 353 ; CHECK: %r = fcmp une <vscale x 4 x float> %a, %b
 354 ; CHECK-NEXT: ret <vscale x 4 x i1> %r
 355   %r = fcmp une <vscale x 4 x float> %a, %b
 356   ret <vscale x 4 x i1> %r
 357 }
 358
 359 define <vscale x 16 x i8> @phi(<vscale x 16 x i8> %a, i32 %val) {
 360 ; CHECK-LABEL: @phi
 361 ; CHECK: %r = phi <vscale x 16 x i8> [ %a, %entry ], [ %added, %iszero ]
 362 ; CHECK-NEXT: ret <vscale x 16 x i8> %r
 363 entry:
 364   %cmp = icmp eq i32 %val, 0
 365   br i1 %cmp, label %iszero, label %end
 366
 367 iszero:
 368   %ins = insertelement <vscale x 16 x i8> undef, i8 1, i32 0
 369   %splatone = shufflevector <vscale x 16 x i8> %ins, <vscale x 16 x i8> undef, <vscale x 16 x i32> zeroinitializer
 370   %added = add <vscale x 16 x i8> %a, %splatone
 371   br label %end
 372
 373 end:
 374   %r = phi <vscale x 16 x i8> [ %a, %entry ], [ %added, %iszero ]
 375   ret <vscale x 16 x i8> %r
 376 }
 377
 378 define <vscale x 8 x half> @select(<vscale x 8 x half> %a, <vscale x 8 x half> %b, <vscale x 8 x i1> %sval) {
 379 ; CHECK-LABEL: @select
 380 ; CHECK: %r = select <vscale x 8 x i1> %sval, <vscale x 8 x half> %a, <vscale x 8 x half> %b
 381 ; CHECK-NEXT: ret <vscale x 8 x half> %r
 382   %r = select <vscale x 8 x i1> %sval, <vscale x 8 x half> %a, <vscale x 8 x half> %b
 383   ret <vscale x 8 x half> %r
 384 }
 385
 386 declare <vscale x 4 x i32> @callee(<vscale x 4 x i32>)
 387 define <vscale x 4 x i32> @call(<vscale x 4 x i32> %val) {
 388 ; CHECK-LABEL: @call
 389 ; CHECK: %r = call <vscale x 4 x i32> @callee(<vscale x 4 x i32> %val)
 390 ; CHECK-NEXT: ret <vscale x 4 x i32> %r
 391   %r = call <vscale x 4 x i32> @callee(<vscale x 4 x i32> %val)
 392   ret <vscale x 4 x i32> %r
 393 }