xref: /llvm-project/llvm/test/Transforms/SLPVectorizer/AArch64/PR38339.ll (revision 580210a0c938531ef9fd79f9ffedb93eeb2e66c2)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2; RUN: opt -passes=slp-vectorizer -S -mtriple=aarch64-apple-ios -mcpu=cyclone -o - %s | FileCheck %s
3
4define void @f1(<2 x i16> %x, ptr %a) {
5; CHECK-LABEL: @f1(
6; CHECK-NEXT:    [[SHUFFLE:%.*]] = shufflevector <2 x i16> [[X:%.*]], <2 x i16> poison, <4 x i32> <i32 0, i32 1, i32 1, i32 0>
7; CHECK-NEXT:    [[TMP1:%.*]] = extractelement <2 x i16> [[X]], i32 0
8; CHECK-NEXT:    store i16 [[TMP1]], ptr [[A:%.*]], align 2
9; CHECK-NEXT:    store <4 x i16> [[SHUFFLE]], ptr undef, align 2
10; CHECK-NEXT:    ret void
11;
12  %t2 = extractelement <2 x i16> %x, i32 0
13  %t3 = extractelement <2 x i16> %x, i32 1
14  %ptr1 = getelementptr inbounds [4 x i16], ptr undef, i16 0, i16 1
15  %ptr2 = getelementptr inbounds [4 x i16], ptr undef, i16 0, i16 2
16  %ptr3 = getelementptr inbounds [4 x i16], ptr undef, i16 0, i16 3
17  store i16 %t2, ptr %a
18  store i16 %t2, ptr undef
19  store i16 %t3, ptr %ptr1
20  store i16 %t3, ptr %ptr2
21  store i16 %t2, ptr %ptr3
22  ret void
23}
24
25define void @f2(<2 x i16> %x, ptr %a) {
26; CHECK-LABEL: @f2(
27; CHECK-NEXT:  entry:
28; CHECK-NEXT:    br label [[CONT:%.*]]
29; CHECK:       cont:
30; CHECK-NEXT:    [[XX:%.*]] = phi <2 x i16> [ [[X:%.*]], [[ENTRY:%.*]] ], [ undef, [[CONT]] ]
31; CHECK-NEXT:    [[AA:%.*]] = phi ptr [ [[A:%.*]], [[ENTRY]] ], [ undef, [[CONT]] ]
32; CHECK-NEXT:    [[SHUFFLE:%.*]] = shufflevector <2 x i16> [[XX]], <2 x i16> poison, <4 x i32> <i32 0, i32 1, i32 1, i32 0>
33; CHECK-NEXT:    [[TMP0:%.*]] = extractelement <2 x i16> [[XX]], i32 0
34; CHECK-NEXT:    store i16 [[TMP0]], ptr [[A]], align 2
35; CHECK-NEXT:    store <4 x i16> [[SHUFFLE]], ptr undef, align 2
36; CHECK-NEXT:    [[A_VAL:%.*]] = load i16, ptr [[A]], align 2
37; CHECK-NEXT:    [[CMP:%.*]] = icmp eq i16 [[A_VAL]], 0
38; CHECK-NEXT:    br i1 [[CMP]], label [[CONT]], label [[EXIT:%.*]]
39; CHECK:       exit:
40; CHECK-NEXT:    ret void
41;
42entry:
43  br label %cont
44
45cont:                                           ; preds = %entry, %cont
46  %xx = phi <2 x i16> [ %x, %entry ], [ undef, %cont ]
47  %aa = phi ptr [ %a, %entry ], [ undef, %cont ]
48  %t2 = extractelement <2 x i16> %xx, i32 0
49  %t3 = extractelement <2 x i16> %xx, i32 1
50  %ptr1 = getelementptr inbounds [4 x i16], ptr undef, i16 0, i16 1
51  %ptr2 = getelementptr inbounds [4 x i16], ptr undef, i16 0, i16 2
52  %ptr3 = getelementptr inbounds [4 x i16], ptr undef, i16 0, i16 3
53  store i16 %t2, ptr %a
54  store i16 %t2, ptr undef
55  store i16 %t3, ptr %ptr1
56  store i16 %t3, ptr %ptr2
57  store i16 %t2, ptr %ptr3
58  %a_val = load i16, ptr %a, align 2
59  %cmp = icmp eq i16 %a_val, 0
60  br i1 %cmp, label %cont, label %exit
61
62exit:                                           ; preds = %cont
63  ret void
64}
65
66define void @f3(<2 x i16> %x, ptr %a) {
67; CHECK-LABEL: @f3(
68; CHECK-NEXT:  entry:
69; CHECK-NEXT:    br label [[CONT:%.*]]
70; CHECK:       cont:
71; CHECK-NEXT:    [[XX:%.*]] = phi <2 x i16> [ [[X:%.*]], [[ENTRY:%.*]] ], [ undef, [[CONT]] ]
72; CHECK-NEXT:    [[AA:%.*]] = phi ptr [ [[A:%.*]], [[ENTRY]] ], [ undef, [[CONT]] ]
73; CHECK-NEXT:    [[SHUFFLE:%.*]] = shufflevector <2 x i16> [[XX]], <2 x i16> poison, <4 x i32> <i32 1, i32 0, i32 0, i32 1>
74; CHECK-NEXT:    [[TMP0:%.*]] = extractelement <2 x i16> [[XX]], i32 1
75; CHECK-NEXT:    store i16 [[TMP0]], ptr [[A]], align 2
76; CHECK-NEXT:    store <4 x i16> [[SHUFFLE]], ptr undef, align 2
77; CHECK-NEXT:    [[A_VAL:%.*]] = load i16, ptr [[A]], align 2
78; CHECK-NEXT:    [[CMP:%.*]] = icmp eq i16 [[A_VAL]], 0
79; CHECK-NEXT:    br i1 [[CMP]], label [[CONT]], label [[EXIT:%.*]]
80; CHECK:       exit:
81; CHECK-NEXT:    ret void
82;
83entry:
84  br label %cont
85
86cont:                                           ; preds = %entry, %cont
87  %xx = phi <2 x i16> [ %x, %entry ], [ undef, %cont ]
88  %aa = phi ptr [ %a, %entry ], [ undef, %cont ]
89  %t2 = extractelement <2 x i16> %xx, i32 0
90  %t3 = extractelement <2 x i16> %xx, i32 1
91  %ptr1 = getelementptr inbounds [4 x i16], ptr undef, i16 0, i16 1
92  %ptr2 = getelementptr inbounds [4 x i16], ptr undef, i16 0, i16 2
93  %ptr3 = getelementptr inbounds [4 x i16], ptr undef, i16 0, i16 3
94  store i16 %t3, ptr %a
95  store i16 %t3, ptr undef
96  store i16 %t2, ptr %ptr1
97  store i16 %t2, ptr %ptr2
98  store i16 %t3, ptr %ptr3
99  %a_val = load i16, ptr %a, align 2
100  %cmp = icmp eq i16 %a_val, 0
101  br i1 %cmp, label %cont, label %exit
102
103exit:                                           ; preds = %cont
104  ret void
105}
106