llvm/test/Transforms/SLPVectorizer/AArch64/buildvector-reduce.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt -S -passes=slp-vectorizer < %s -mtriple=arm64-apple-macosx | FileCheck %s
   3 ; RUN: opt -S -passes=slp-vectorizer < %s -mtriple=arm64-apple-macosx -slp-optimize-identity-hor-reduction-ops=false | FileCheck %s --check-prefix=NO-IDENTITY
   4
   5 define i8 @test() {
   6 ; CHECK-LABEL: @test(
   7 ; CHECK-NEXT:  entry:
   8 ; CHECK-NEXT:    br label [[FOR_BODY:%.*]]
   9 ; CHECK:       for.body:
  10 ; CHECK-NEXT:    [[SUM:%.*]] = phi i32 [ [[TMP0:%.*]], [[FOR_BODY]] ], [ 0, [[ENTRY:%.*]] ]
  11 ; CHECK-NEXT:    [[CALL278:%.*]] = call i32 @fn(i32 [[SUM]])
  12 ; CHECK-NEXT:    [[TMP0]] = mul i32 [[CALL278]], 8
  13 ; CHECK-NEXT:    br label [[FOR_BODY]]
  14 ;
  15 ; NO-IDENTITY-LABEL: @test(
  16 ; NO-IDENTITY-NEXT:  entry:
  17 ; NO-IDENTITY-NEXT:    br label [[FOR_BODY:%.*]]
  18 ; NO-IDENTITY:       for.body:
  19 ; NO-IDENTITY-NEXT:    [[SUM:%.*]] = phi i32 [ [[TMP2:%.*]], [[FOR_BODY]] ], [ 0, [[ENTRY:%.*]] ]
  20 ; NO-IDENTITY-NEXT:    [[CALL278:%.*]] = call i32 @fn(i32 [[SUM]])
  21 ; NO-IDENTITY-NEXT:    [[TMP0:%.*]] = insertelement <8 x i32> poison, i32 [[CALL278]], i32 0
  22 ; NO-IDENTITY-NEXT:    [[TMP1:%.*]] = shufflevector <8 x i32> [[TMP0]], <8 x i32> poison, <8 x i32> zeroinitializer
  23 ; NO-IDENTITY-NEXT:    [[TMP2]] = call i32 @llvm.vector.reduce.add.v8i32(<8 x i32> [[TMP1]])
  24 ; NO-IDENTITY-NEXT:    br label [[FOR_BODY]]
  25 ;
  26 entry:
  27   br label %for.body
  28
  29 for.body:
  30   %sum = phi i32 [ %add285.19, %for.body ], [ 0, %entry ]
  31   %call278 = call i32 @fn(i32 %sum)
  32   %add285.13 = add i32 %call278, %call278
  33   %add285.14 = add i32 %add285.13, %call278
  34   %add285.15 = add i32 %add285.14, %call278
  35   %add285.16 = add i32 %add285.15, %call278
  36   %add285.17 = add i32 %add285.16, %call278
  37   %add285.18 = add i32 %add285.17, %call278
  38   %add285.19 = add i32 %add285.18, %call278
  39   br label %for.body
  40 }
  41
  42 declare i32 @fn(i32)