llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-trunc-stores.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc -mattr=+sve -force-streaming-compatible-sve < %s | FileCheck %s
   3
   4
   5 target triple = "aarch64-unknown-linux-gnu"
   6
   7 define void @store_trunc_v8i16i8(ptr %ap, ptr %dest) {
   8 ; CHECK-LABEL: store_trunc_v8i16i8:
   9 ; CHECK:       // %bb.0:
  10 ; CHECK-NEXT:    ptrue p0.h, vl8
  11 ; CHECK-NEXT:    ldr q0, [x0]
  12 ; CHECK-NEXT:    st1b { z0.h }, p0, [x1]
  13 ; CHECK-NEXT:    ret
  14   %a = load <8 x i16>, ptr %ap
  15   %val = trunc <8 x i16> %a to <8 x i8>
  16   store <8 x i8> %val, ptr %dest
  17   ret void
  18 }
  19
  20 define void @store_trunc_v4i32i8(ptr %ap, ptr %dest) {
  21 ; CHECK-LABEL: store_trunc_v4i32i8:
  22 ; CHECK:       // %bb.0:
  23 ; CHECK-NEXT:    ptrue p0.s, vl4
  24 ; CHECK-NEXT:    ldr q0, [x0]
  25 ; CHECK-NEXT:    st1b { z0.s }, p0, [x1]
  26 ; CHECK-NEXT:    ret
  27   %a = load <4 x i32>, ptr %ap
  28   %val = trunc <4 x i32> %a to <4 x i8>
  29   store <4 x i8> %val, ptr %dest
  30   ret void
  31 }
  32
  33 define void @store_trunc_v4i32i16(ptr %ap, ptr %dest) {
  34 ; CHECK-LABEL: store_trunc_v4i32i16:
  35 ; CHECK:       // %bb.0:
  36 ; CHECK-NEXT:    ptrue p0.s, vl4
  37 ; CHECK-NEXT:    ldr q0, [x0]
  38 ; CHECK-NEXT:    st1h { z0.s }, p0, [x1]
  39 ; CHECK-NEXT:    ret
  40   %a = load <4 x i32>, ptr %ap
  41   %val = trunc <4 x i32> %a to <4 x i16>
  42   store <4 x i16> %val, ptr %dest
  43   ret void
  44 }
  45
  46 define void @store_trunc_v2i64i8(ptr %ap, ptr %dest) {
  47 ; CHECK-LABEL: store_trunc_v2i64i8:
  48 ; CHECK:       // %bb.0:
  49 ; CHECK-NEXT:    ptrue p0.d, vl2
  50 ; CHECK-NEXT:    ldr q0, [x0]
  51 ; CHECK-NEXT:    st1w { z0.d }, p0, [x1]
  52 ; CHECK-NEXT:    ret
  53   %a = load <2 x i64>, ptr %ap
  54   %val = trunc <2 x i64> %a to <2 x i32>
  55   store <2 x i32> %val, ptr %dest
  56   ret void
  57 }
  58
  59 define void @store_trunc_v2i256i64(ptr %ap, ptr %dest) {
  60 ; CHECK-LABEL: store_trunc_v2i256i64:
  61 ; CHECK:       // %bb.0:
  62 ; CHECK-NEXT:    ptrue p0.d, vl1
  63 ; CHECK-NEXT:    ldr d0, [x0, #32]
  64 ; CHECK-NEXT:    ldr d1, [x0]
  65 ; CHECK-NEXT:    splice z1.d, p0, z1.d, z0.d
  66 ; CHECK-NEXT:    str q1, [x1]
  67 ; CHECK-NEXT:    ret
  68   %a = load <2 x i256>, ptr %ap
  69   %val = trunc <2 x i256> %a to <2 x i64>
  70   store <2 x i64> %val, ptr %dest
  71   ret void
  72 }