test/Transforms/ScalarizeMaskedMemIntrin/X86/expand-masked-store.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt -S %s -scalarize-masked-mem-intrin -mtriple=x86_64-linux-gnu | FileCheck %s
   3
   4 define void @scalarize_v2i64(<2 x i64>* %p, <2 x i1> %mask, <2 x i64> %data) {
   5 ; CHECK-LABEL: @scalarize_v2i64(
   6 ; CHECK-NEXT:    [[TMP1:%.*]] = bitcast <2 x i64>* [[P:%.*]] to i64*
   7 ; CHECK-NEXT:    [[TMP2:%.*]] = extractelement <2 x i1> [[MASK:%.*]], i64 0
   8 ; CHECK-NEXT:    br i1 [[TMP2]], label [[COND_STORE:%.*]], label [[ELSE:%.*]]
   9 ; CHECK:       cond.store:
  10 ; CHECK-NEXT:    [[TMP3:%.*]] = extractelement <2 x i64> [[DATA:%.*]], i64 0
  11 ; CHECK-NEXT:    [[TMP4:%.*]] = getelementptr inbounds i64, i64* [[TMP1]], i32 0
  12 ; CHECK-NEXT:    store i64 [[TMP3]], i64* [[TMP4]], align 8
  13 ; CHECK-NEXT:    br label [[ELSE]]
  14 ; CHECK:       else:
  15 ; CHECK-NEXT:    [[TMP5:%.*]] = extractelement <2 x i1> [[MASK]], i64 1
  16 ; CHECK-NEXT:    br i1 [[TMP5]], label [[COND_STORE1:%.*]], label [[ELSE2:%.*]]
  17 ; CHECK:       cond.store1:
  18 ; CHECK-NEXT:    [[TMP6:%.*]] = extractelement <2 x i64> [[DATA]], i64 1
  19 ; CHECK-NEXT:    [[TMP7:%.*]] = getelementptr inbounds i64, i64* [[TMP1]], i32 1
  20 ; CHECK-NEXT:    store i64 [[TMP6]], i64* [[TMP7]], align 8
  21 ; CHECK-NEXT:    br label [[ELSE2]]
  22 ; CHECK:       else2:
  23 ; CHECK-NEXT:    ret void
  24 ;
  25   call void @llvm.masked.store.v2i64.p0v2i64(<2 x i64> %data, <2 x i64>* %p, i32 128, <2 x i1> %mask)
  26   ret void
  27 }
  28
  29 define void @scalarize_v2i64_ones_mask(<2 x i64>* %p, <2 x i64> %data) {
  30 ; CHECK-LABEL: @scalarize_v2i64_ones_mask(
  31 ; CHECK-NEXT:    store <2 x i64> [[DATA:%.*]], <2 x i64>* [[P:%.*]], align 8
  32 ; CHECK-NEXT:    ret void
  33 ;
  34   call void @llvm.masked.store.v2i64.p0v2i64(<2 x i64> %data, <2 x i64>* %p, i32 8, <2 x i1> <i1 true, i1 true>)
  35   ret void
  36 }
  37
  38 define void @scalarize_v2i64_zero_mask(<2 x i64>* %p, <2 x i64> %data) {
  39 ; CHECK-LABEL: @scalarize_v2i64_zero_mask(
  40 ; CHECK-NEXT:    [[TMP1:%.*]] = bitcast <2 x i64>* [[P:%.*]] to i64*
  41 ; CHECK-NEXT:    ret void
  42 ;
  43   call void @llvm.masked.store.v2i64.p0v2i64(<2 x i64> %data, <2 x i64>* %p, i32 8, <2 x i1> <i1 false, i1 false>)
  44   ret void
  45 }
  46
  47 define void @scalarize_v2i64_const_mask(<2 x i64>* %p, <2 x i64> %data) {
  48 ; CHECK-LABEL: @scalarize_v2i64_const_mask(
  49 ; CHECK-NEXT:    [[TMP1:%.*]] = bitcast <2 x i64>* [[P:%.*]] to i64*
  50 ; CHECK-NEXT:    [[TMP2:%.*]] = extractelement <2 x i64> [[DATA:%.*]], i64 1
  51 ; CHECK-NEXT:    [[TMP3:%.*]] = getelementptr inbounds i64, i64* [[TMP1]], i32 1
  52 ; CHECK-NEXT:    store i64 [[TMP2]], i64* [[TMP3]], align 8
  53 ; CHECK-NEXT:    ret void
  54 ;
  55   call void @llvm.masked.store.v2i64.p0v2i64(<2 x i64> %data, <2 x i64>* %p, i32 8, <2 x i1> <i1 false, i1 true>)
  56   ret void
  57 }
  58
  59 declare void @llvm.masked.store.v2i64.p0v2i64(<2 x i64>, <2 x i64>*, i32, <2 x i1>)