xref: /llvm-project/llvm/test/Transforms/ScalarizeMaskedMemIntrin/X86/expand-masked-scatter.ll (revision e417a7d1223cc47bf94def379b69772548a9bb04)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2; RUN: opt -S %s -passes=scalarize-masked-mem-intrin -mtriple=x86_64-linux-gnu | FileCheck %s
3
4define void @scalarize_v2i64(<2 x ptr> %p, <2 x i1> %mask, <2 x i64> %value) {
5; CHECK-LABEL: @scalarize_v2i64(
6; CHECK-NEXT:    [[SCALAR_MASK:%.*]] = bitcast <2 x i1> [[MASK:%.*]] to i2
7; CHECK-NEXT:    [[TMP1:%.*]] = and i2 [[SCALAR_MASK]], 1
8; CHECK-NEXT:    [[TMP2:%.*]] = icmp ne i2 [[TMP1]], 0
9; CHECK-NEXT:    br i1 [[TMP2]], label [[COND_STORE:%.*]], label [[ELSE:%.*]]
10; CHECK:       cond.store:
11; CHECK-NEXT:    [[ELT0:%.*]] = extractelement <2 x i64> [[VALUE:%.*]], i64 0
12; CHECK-NEXT:    [[PTR0:%.*]] = extractelement <2 x ptr> [[P:%.*]], i64 0
13; CHECK-NEXT:    store i64 [[ELT0]], ptr [[PTR0]], align 8
14; CHECK-NEXT:    br label [[ELSE]]
15; CHECK:       else:
16; CHECK-NEXT:    [[TMP3:%.*]] = and i2 [[SCALAR_MASK]], -2
17; CHECK-NEXT:    [[TMP4:%.*]] = icmp ne i2 [[TMP3]], 0
18; CHECK-NEXT:    br i1 [[TMP4]], label [[COND_STORE1:%.*]], label [[ELSE2:%.*]]
19; CHECK:       cond.store1:
20; CHECK-NEXT:    [[ELT1:%.*]] = extractelement <2 x i64> [[VALUE]], i64 1
21; CHECK-NEXT:    [[PTR1:%.*]] = extractelement <2 x ptr> [[P]], i64 1
22; CHECK-NEXT:    store i64 [[ELT1]], ptr [[PTR1]], align 8
23; CHECK-NEXT:    br label [[ELSE2]]
24; CHECK:       else2:
25; CHECK-NEXT:    ret void
26;
27  call void @llvm.masked.scatter.v2i64.v2p0(<2 x i64> %value, <2 x ptr> %p, i32 8, <2 x i1> %mask)
28  ret void
29}
30
31define void @scalarize_v2i64_ones_mask(<2 x ptr> %p, <2 x i64> %value) {
32; CHECK-LABEL: @scalarize_v2i64_ones_mask(
33; CHECK-NEXT:    [[ELT0:%.*]] = extractelement <2 x i64> [[VALUE:%.*]], i64 0
34; CHECK-NEXT:    [[PTR0:%.*]] = extractelement <2 x ptr> [[P:%.*]], i64 0
35; CHECK-NEXT:    store i64 [[ELT0]], ptr [[PTR0]], align 8
36; CHECK-NEXT:    [[ELT1:%.*]] = extractelement <2 x i64> [[VALUE]], i64 1
37; CHECK-NEXT:    [[PTR1:%.*]] = extractelement <2 x ptr> [[P]], i64 1
38; CHECK-NEXT:    store i64 [[ELT1]], ptr [[PTR1]], align 8
39; CHECK-NEXT:    ret void
40;
41  call void @llvm.masked.scatter.v2i64.v2p0(<2 x i64> %value, <2 x ptr> %p, i32 8, <2 x i1> <i1 true, i1 true>)
42  ret void
43}
44
45define void @scalarize_v2i64_zero_mask(<2 x ptr> %p, <2 x i64> %value) {
46; CHECK-LABEL: @scalarize_v2i64_zero_mask(
47; CHECK-NEXT:    ret void
48;
49  call void @llvm.masked.scatter.v2i64.v2p0(<2 x i64> %value, <2 x ptr> %p, i32 8, <2 x i1> <i1 false, i1 false>)
50  ret void
51}
52
53define void @scalarize_v2i64_const_mask(<2 x ptr> %p, <2 x i64> %value) {
54; CHECK-LABEL: @scalarize_v2i64_const_mask(
55; CHECK-NEXT:    [[ELT1:%.*]] = extractelement <2 x i64> [[VALUE:%.*]], i64 1
56; CHECK-NEXT:    [[PTR1:%.*]] = extractelement <2 x ptr> [[P:%.*]], i64 1
57; CHECK-NEXT:    store i64 [[ELT1]], ptr [[PTR1]], align 8
58; CHECK-NEXT:    ret void
59;
60  call void @llvm.masked.scatter.v2i64.v2p0(<2 x i64> %value, <2 x ptr> %p, i32 8, <2 x i1> <i1 false, i1 true>)
61  ret void
62}
63
64declare void @llvm.masked.scatter.v2i64.v2p0(<2 x i64>, <2 x ptr>, i32, <2 x i1>)
65