[llvm] r295050 - [SLP] Additional tests for extractelement cost fix.

[Alexey Bataev via llvm-commits]

Author: abataev
Date: Tue Feb 14 06:52:05 2017
New Revision: 295050

URL: http://llvm.org/viewvc/llvm-project?rev=295050&view=rev
Log:
[SLP] Additional tests for extractelement cost fix.

Modified:
    llvm/trunk/test/Transforms/SLPVectorizer/X86/extractelement.ll

Modified: llvm/trunk/test/Transforms/SLPVectorizer/X86/extractelement.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/Transforms/SLPVectorizer/X86/extractelement.ll?rev=295050&r1=295049&r2=295050&view=diff
==============================================================================
--- llvm/trunk/test/Transforms/SLPVectorizer/X86/extractelement.ll (original)
+++ llvm/trunk/test/Transforms/SLPVectorizer/X86/extractelement.ll Tue Feb 14 06:52:05 2017
@@ -1,5 +1,9 @@
 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
 ; RUN: opt < %s -slp-vectorizer -S -mtriple=x86_64-unknown-linux -march=core-avx2 | FileCheck %s
+; RUN: opt < %s -slp-vectorizer -S -mtriple=x86_64-unknown-linux -march=core-avx2 -slp-threshold=-1 -slp-vectorize-hor-store | FileCheck %s --check-prefix=THRESH1
+; RUN: opt < %s -slp-vectorizer -S -mtriple=x86_64-unknown-linux -march=core-avx2 -slp-threshold=-2 -slp-vectorize-hor-store | FileCheck %s --check-prefix=THRESH2
+
+ at a = global float 0.000000e+00, align 4
 
 define float @f(<2 x float> %x) {
 ; CHECK-LABEL: @f(
@@ -16,5 +20,40 @@ define float @f(<2 x float> %x) {
   %x1x1 = fmul float %x1, %x1
   %add = fadd float %x0x0, %x1x1
   ret float %add
+}
+
+define float @f_used_out_of_tree(<2 x float> %x) {
+; THRESH2-LABEL: @f_used_out_of_tree(
+; THRESH2-NEXT:    [[X0:%.*]] = extractelement <2 x float> [[X:%.*]], i32 0
+; THRESH2-NEXT:    [[X1:%.*]] = extractelement <2 x float> [[X]], i32 1
+; THRESH2-NEXT:    [[X0X0:%.*]] = fmul float [[X0]], [[X0]]
+; THRESH2-NEXT:    [[X1X1:%.*]] = fmul float [[X1]], [[X1]]
+; THRESH2-NEXT:    [[ADD:%.*]] = fadd float [[X0X0]], [[X1X1]]
+; THRESH2-NEXT:    store float [[ADD]], float* @a
+; THRESH2-NEXT:    ret float [[X0]]
+;
+  %x0 = extractelement <2 x float> %x, i32 0
+  %x1 = extractelement <2 x float> %x, i32 1
+  %x0x0 = fmul float %x0, %x0
+  %x1x1 = fmul float %x1, %x1
+  %add = fadd float %x0x0, %x1x1
+  store float %add, float* @a
+  ret float %x0
+}
+
+define float @f_used_twice_in_tree(<2 x float> %x) {
+; THRESH1-LABEL: @f_used_twice_in_tree(
+; THRESH1-NEXT:    [[X0:%.*]] = extractelement <2 x float> [[X:%.*]], i32 0
+; THRESH1-NEXT:    [[X1:%.*]] = extractelement <2 x float> [[X]], i32 1
+; THRESH1-NEXT:    [[X0X0:%.*]] = fmul float [[X0]], [[X1]]
+; THRESH1-NEXT:    [[X1X1:%.*]] = fmul float [[X1]], [[X1]]
+; THRESH1-NEXT:    [[ADD:%.*]] = fadd float [[X0X0]], [[X1X1]]
+; THRESH1-NEXT:    ret float [[ADD]]
+  %x0 = extractelement <2 x float> %x, i32 0
+  %x1 = extractelement <2 x float> %x, i32 1
+  %x0x0 = fmul float %x0, %x1
+  %x1x1 = fmul float %x1, %x1
+  %add = fadd float %x0x0, %x1x1
+  ret float %add
 }