llvm/test/CodeGen/AArch64/sve-dead-masked-store.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
   4 define void @dead_masked_store(<vscale x 4 x i32> %val, ptr %a, <vscale x 4 x i1> %mask) {
   5 ; CHECK-LABEL: dead_masked_store:
   6 ; CHECK:       // %bb.0:
   7 ; CHECK-NEXT:    st1w { z0.s }, p0, [x0]
   8 ; CHECK-NEXT:    ret
   9   call void @llvm.masked.store.nxv4i32(<vscale x 4 x i32> %val, ptr %a, i32 4, <vscale x 4 x i1> %mask)
  10   call void @llvm.masked.store.nxv4i32(<vscale x 4 x i32> %val, ptr %a, i32 4, <vscale x 4 x i1> %mask)
  11   ret void
  12 }
  13
  14 define void @dead_masked_store_alltrue_same(<vscale x 4 x i32> %val, ptr %a, <vscale x 4 x i1> %mask) {
  15 ; CHECK-LABEL: dead_masked_store_alltrue_same:
  16 ; CHECK:       // %bb.0:
  17 ; CHECK-NEXT:    ptrue p0.s
  18 ; CHECK-NEXT:    st1w { z0.s }, p0, [x0]
  19 ; CHECK-NEXT:    ret
  20   %alltrue.ins = insertelement <vscale x 4 x i1> poison, i1 true, i32 0
  21   %alltrue = shufflevector  <vscale x 4 x i1> %alltrue.ins,  <vscale x 4 x i1> poison,  <vscale x 4 x i32> zeroinitializer
  22   call void @llvm.masked.store.nxv4i32(<vscale x 4 x i32> %val, ptr %a, i32 4, <vscale x 4 x i1> %mask)
  23   call void @llvm.masked.store.nxv4i32(<vscale x 4 x i32> %val, ptr %a, i32 4, <vscale x 4 x i1> %alltrue)
  24   ret void
  25 }
  26
  27 define void @dead_masked_store_alltrue_bigger(<vscale x 4 x i16> %val, <vscale x 4 x i32> %val1, ptr %a, <vscale x 4 x i1> %mask) {
  28 ; CHECK-LABEL: dead_masked_store_alltrue_bigger:
  29 ; CHECK:       // %bb.0:
  30 ; CHECK-NEXT:    ptrue p0.s
  31 ; CHECK-NEXT:    st1w { z1.s }, p0, [x0]
  32 ; CHECK-NEXT:    ret
  33   %alltrue.ins = insertelement <vscale x 4 x i1> poison, i1 true, i32 0
  34   %alltrue = shufflevector  <vscale x 4 x i1> %alltrue.ins,  <vscale x 4 x i1> poison,  <vscale x 4 x i32> zeroinitializer
  35   call void @llvm.masked.store.nxv4i16(<vscale x 4 x i16> %val, ptr %a, i32 4, <vscale x 4 x i1> %mask)
  36   call void @llvm.masked.store.nxv4i32(<vscale x 4 x i32> %val1, ptr %a, i32 4, <vscale x 4 x i1> %alltrue)
  37   ret void
  38 }
  39
  40 define void @dead_masked_store_alltrue_smaller(<vscale x 4 x i32> %val, <vscale x 4 x i16> %val1, ptr %a, <vscale x 4 x i1> %mask) {
  41 ; CHECK-LABEL: dead_masked_store_alltrue_smaller:
  42 ; CHECK:       // %bb.0:
  43 ; CHECK-NEXT:    ptrue p1.s
  44 ; CHECK-NEXT:    st1w { z0.s }, p0, [x0]
  45 ; CHECK-NEXT:    st1h { z1.s }, p1, [x0]
  46 ; CHECK-NEXT:    ret
  47   %alltrue.ins = insertelement <vscale x 4 x i1> poison, i1 true, i32 0
  48   %alltrue = shufflevector  <vscale x 4 x i1> %alltrue.ins,  <vscale x 4 x i1> poison,  <vscale x 4 x i32> zeroinitializer
  49   call void @llvm.masked.store.nxv4i32(<vscale x 4 x i32> %val, ptr %a, i32 4, <vscale x 4 x i1> %mask)
  50   call void @llvm.masked.store.nxv4i16(<vscale x 4 x i16> %val1, ptr %a, i32 4, <vscale x 4 x i1> %alltrue)
  51   ret void
  52 }
  53
  54 define void @dead_masked_store_same_mask_smaller_type(<vscale x 4 x i32> %val, <vscale x 4 x i16> %val1, ptr %a, <vscale x 4 x i1> %mask) {
  55 ; CHECK-LABEL: dead_masked_store_same_mask_smaller_type:
  56 ; CHECK:       // %bb.0:
  57 ; CHECK-NEXT:    st1w { z0.s }, p0, [x0]
  58 ; CHECK-NEXT:    st1h { z1.s }, p0, [x0]
  59 ; CHECK-NEXT:    ret
  60   call void @llvm.masked.store.nxv4i32(<vscale x 4 x i32> %val, ptr %a, i32 4, <vscale x 4 x i1> %mask)
  61   call void @llvm.masked.store.nxv4i16(<vscale x 4 x i16> %val1, ptr %a, i32 4, <vscale x 4 x i1> %mask)
  62   ret void
  63 }
  64
  65 define void @dead_masked_store_same_mask_bigger_type(<vscale x 4 x i16> %val, <vscale x 4 x i32> %val1, ptr %a, <vscale x 4 x i1> %mask) {
  66 ; CHECK-LABEL: dead_masked_store_same_mask_bigger_type:
  67 ; CHECK:       // %bb.0:
  68 ; CHECK-NEXT:    st1h { z0.s }, p0, [x0]
  69 ; CHECK-NEXT:    st1w { z1.s }, p0, [x0]
  70 ; CHECK-NEXT:    ret
  71   call void @llvm.masked.store.nxv4i16(<vscale x 4 x i16> %val, ptr %a, i32 4, <vscale x 4 x i1> %mask)
  72   call void @llvm.masked.store.nxv4i32(<vscale x 4 x i32> %val1, ptr %a, i32 4, <vscale x 4 x i1> %mask)
  73   ret void
  74 }
  75
  76 declare void @llvm.masked.store.nxv4i16(<vscale x 4 x i16>, ptr, i32, <vscale x 4 x i1>)
  77 declare void @llvm.masked.store.nxv4i32(<vscale x 4 x i32>, ptr, i32, <vscale x 4 x i1>)