xref: /llvm-project/llvm/test/Transforms/SLPVectorizer/diamond_broadcast.ll (revision e0bd8d3485075d24ecff2b4f5d9e2117853bd08b)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2; RUN: %if x86-registered-target %{ opt < %s -passes=slp-vectorizer -S -mtriple=x86_64-unknown-linux -slp-threshold=-1 | FileCheck %s %}
3; RUN: %if aarch64-registered-target %{ opt < %s -passes=slp-vectorizer -S -mtriple=aarch64-unknown-linux-gnu -slp-threshold=-1 | FileCheck %s %}
4
5define i32 @diamond_broadcast(ptr noalias nocapture %B, ptr noalias nocapture %A) {
6; CHECK-LABEL: @diamond_broadcast(
7; CHECK-NEXT:  entry:
8; CHECK-NEXT:    [[LD:%.*]] = load i32, ptr [[A:%.*]], align 4
9; CHECK-NEXT:    [[TMP0:%.*]] = insertelement <4 x i32> poison, i32 [[LD]], i32 0
10; CHECK-NEXT:    [[SHUFFLE:%.*]] = shufflevector <4 x i32> [[TMP0]], <4 x i32> poison, <4 x i32> zeroinitializer
11; CHECK-NEXT:    [[TMP1:%.*]] = mul <4 x i32> [[SHUFFLE]], [[SHUFFLE]]
12; CHECK-NEXT:    store <4 x i32> [[TMP1]], ptr [[B:%.*]], align 4
13; CHECK-NEXT:    ret i32 0
14;
15entry:
16  %ld = load i32, ptr %A, align 4
17  %mul = mul i32 %ld, %ld
18  store i32 %mul, ptr %B, align 4
19  %mul8 = mul i32 %ld, %ld
20  %arrayidx9 = getelementptr inbounds i32, ptr %B, i64 1
21  store i32 %mul8, ptr %arrayidx9, align 4
22  %mul14 = mul i32 %ld, %ld
23  %arrayidx15 = getelementptr inbounds i32, ptr %B, i64 2
24  store i32 %mul14, ptr %arrayidx15, align 4
25  %mul20 = mul i32 %ld, %ld
26  %arrayidx21 = getelementptr inbounds i32, ptr %B, i64 3
27  store i32 %mul20, ptr %arrayidx21, align 4
28  ret i32 0
29}
30
31
32