xref: /llvm-project/llvm/test/Transforms/LoopVectorize/X86/pr42674.ll (revision 10f315dc9c96ec2413881ab55a285e35d80def88)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2; RUN: opt %s -passes=loop-vectorize,instcombine,simplifycfg -simplifycfg-require-and-preserve-domtree=1 -mtriple=x86_64-unknown-linux-gnu -mattr=avx512vl,avx512dq,avx512bw -S | FileCheck %s
3
4@bytes = global [128 x i8] zeroinitializer, align 16
5
6; Make sure we end up with vector code for this loop. We used to try to create
7; a VF=64,UF=4 loop, but the scalar trip count is only 128 so
8; the vector loop was dead code leaving only a scalar remainder.
9define zeroext i8 @sum() {
10; CHECK-LABEL: @sum(
11; CHECK-NEXT:  entry:
12; CHECK-NEXT:    [[WIDE_LOAD2:%.*]] = load <64 x i8>, ptr getelementptr inbounds nuw (i8, ptr @bytes, i64 64), align 1
13; CHECK-NEXT:    [[WIDE_LOAD:%.*]] = load <64 x i8>, ptr @bytes, align 1
14; CHECK-NEXT:    [[BIN_RDX:%.*]] = add <64 x i8> [[WIDE_LOAD2]], [[WIDE_LOAD]]
15; CHECK-NEXT:    [[TMP0:%.*]] = call i8 @llvm.vector.reduce.add.v64i8(<64 x i8> [[BIN_RDX]])
16; CHECK-NEXT:    ret i8 [[TMP0]]
17;
18entry:
19  br label %for.body
20
21for.body:                                         ; preds = %for.body, %entry
22  %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ]
23  %r.010 = phi i8 [ 0, %entry ], [ %add, %for.body ]
24  %arrayidx = getelementptr inbounds [128 x i8], ptr @bytes, i64 0, i64 %indvars.iv
25  %0 = load i8, ptr %arrayidx, align 1
26  %add = add i8 %0, %r.010
27  %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
28  %exitcond = icmp eq i64 %indvars.iv.next, 128
29  br i1 %exitcond, label %for.end, label %for.body
30
31for.end:                                          ; preds = %for.body
32  %add.lcssa = phi i8 [ %add, %for.body ]
33  ret i8 %add.lcssa
34}
35