xref: /llvm-project/llvm/test/Transforms/SLPVectorizer/X86/diamond_broadcast_extra_shuffle.ll (revision 580210a0c938531ef9fd79f9ffedb93eeb2e66c2)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2; RUN: opt < %s -passes=slp-vectorizer -S -mtriple=x86_64-unknown-linux -slp-threshold=-2 | FileCheck %s
3
4define i32 @diamond_broadcast(ptr noalias nocapture %B, ptr noalias nocapture %A) {
5; CHECK-LABEL: @diamond_broadcast(
6; CHECK-NEXT:  entry:
7; CHECK-NEXT:    [[LD:%.*]] = load i32, ptr [[A:%.*]], align 4
8; CHECK-NEXT:    [[TMP0:%.*]] = insertelement <4 x i32> poison, i32 [[LD]], i32 0
9; CHECK-NEXT:    [[SHUFFLE:%.*]] = shufflevector <4 x i32> [[TMP0]], <4 x i32> poison, <4 x i32> zeroinitializer
10; CHECK-NEXT:    [[TMP1:%.*]] = mul <4 x i32> [[SHUFFLE]], [[SHUFFLE]]
11; CHECK-NEXT:    store <4 x i32> [[TMP1]], ptr [[B:%.*]], align 4
12; CHECK-NEXT:    ret i32 0
13;
14entry:
15  %ld = load i32, ptr %A, align 4
16  %mul = mul i32 %ld, %ld
17  store i32 %mul, ptr %B, align 4
18  %mul8 = mul i32 %ld, %ld
19  %arrayidx9 = getelementptr inbounds i32, ptr %B, i64 1
20  store i32 %mul8, ptr %arrayidx9, align 4
21  %mul14 = mul i32 %ld, %ld
22  %arrayidx15 = getelementptr inbounds i32, ptr %B, i64 2
23  store i32 %mul14, ptr %arrayidx15, align 4
24  %mul20 = mul i32 %ld, undef
25  %arrayidx21 = getelementptr inbounds i32, ptr %B, i64 3
26  store i32 %mul20, ptr %arrayidx21, align 4
27  ret i32 0
28}
29
30define i32 @diamond_broadcast2(ptr noalias nocapture %B, ptr noalias nocapture %A) {
31; CHECK-LABEL: @diamond_broadcast2(
32; CHECK-NEXT:  entry:
33; CHECK-NEXT:    [[LD:%.*]] = load i32, ptr [[A:%.*]], align 4
34; CHECK-NEXT:    [[TMP0:%.*]] = insertelement <4 x i32> poison, i32 [[LD]], i32 0
35; CHECK-NEXT:    [[SHUFFLE:%.*]] = shufflevector <4 x i32> [[TMP0]], <4 x i32> poison, <4 x i32> zeroinitializer
36; CHECK-NEXT:    [[TMP1:%.*]] = mul <4 x i32> [[SHUFFLE]], [[SHUFFLE]]
37; CHECK-NEXT:    store <4 x i32> [[TMP1]], ptr [[B:%.*]], align 4
38; CHECK-NEXT:    ret i32 0
39;
40entry:
41  %ld = load i32, ptr %A, align 4
42  %mul = mul i32 %ld, %ld
43  store i32 %mul, ptr %B, align 4
44  %mul8 = mul i32 %ld, %ld
45  %arrayidx9 = getelementptr inbounds i32, ptr %B, i64 1
46  store i32 %mul8, ptr %arrayidx9, align 4
47  %mul14 = mul i32 %ld, %ld
48  %arrayidx15 = getelementptr inbounds i32, ptr %B, i64 2
49  store i32 %mul14, ptr %arrayidx15, align 4
50  %mul20 = mul i32 undef, %ld
51  %arrayidx21 = getelementptr inbounds i32, ptr %B, i64 3
52  store i32 %mul20, ptr %arrayidx21, align 4
53  ret i32 0
54}
55
56define i32 @diamond_broadcast3(ptr noalias nocapture %B, ptr noalias nocapture %A) {
57; CHECK-LABEL: @diamond_broadcast3(
58; CHECK-NEXT:  entry:
59; CHECK-NEXT:    [[LD:%.*]] = load i32, ptr [[A:%.*]], align 4
60; CHECK-NEXT:    [[TMP0:%.*]] = insertelement <4 x i32> poison, i32 [[LD]], i32 0
61; CHECK-NEXT:    [[SHUFFLE:%.*]] = shufflevector <4 x i32> [[TMP0]], <4 x i32> poison, <4 x i32> zeroinitializer
62; CHECK-NEXT:    [[TMP1:%.*]] = mul <4 x i32> [[SHUFFLE]], [[SHUFFLE]]
63; CHECK-NEXT:    store <4 x i32> [[TMP1]], ptr [[B:%.*]], align 4
64; CHECK-NEXT:    ret i32 0
65;
66entry:
67  %ld = load i32, ptr %A, align 4
68  %mul = mul i32 %ld, %ld
69  store i32 %mul, ptr %B, align 4
70  %mul8 = mul i32 %ld, %ld
71  %arrayidx9 = getelementptr inbounds i32, ptr %B, i64 1
72  store i32 %mul8, ptr %arrayidx9, align 4
73  %mul14 = mul i32 %ld, undef
74  %arrayidx15 = getelementptr inbounds i32, ptr %B, i64 2
75  store i32 %mul14, ptr %arrayidx15, align 4
76  %mul20 = mul i32 undef, %ld
77  %arrayidx21 = getelementptr inbounds i32, ptr %B, i64 3
78  store i32 %mul20, ptr %arrayidx21, align 4
79  ret i32 0
80}
81
82