llvm/test/Transforms/SLPVectorizer/X86/vec3-calls.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt -passes=slp-vectorizer -slp-vectorize-non-power-of-2 -mtriple=x86_64-apple-macosx -S %s | FileCheck --check-prefixes=CHECK,NON-POW2 %s
   3 ; RUN: opt -passes=slp-vectorizer -slp-vectorize-non-power-of-2=false -mtriple=x86_64-apple-macosx -S %s | FileCheck --check-prefixes=CHECK,POW2-ONLY %s
   4
   5 define void @vec3_vectorize_call(ptr %Colour, float %0) {
   6 ; NON-POW2-LABEL: @vec3_vectorize_call(
   7 ; NON-POW2-NEXT:  entry:
   8 ; NON-POW2-NEXT:    [[TMP1:%.*]] = load float, ptr [[COLOUR:%.*]], align 4
   9 ; NON-POW2-NEXT:    [[ARRAYIDX91_I:%.*]] = getelementptr float, ptr [[COLOUR]], i64 1
  10 ; NON-POW2-NEXT:    [[TMP2:%.*]] = load float, ptr [[ARRAYIDX91_I]], align 4
  11 ; NON-POW2-NEXT:    [[TMP3:%.*]] = insertelement <3 x float> poison, float [[TMP0:%.*]], i32 2
  12 ; NON-POW2-NEXT:    [[TMP4:%.*]] = insertelement <3 x float> [[TMP3]], float [[TMP1]], i32 0
  13 ; NON-POW2-NEXT:    [[TMP5:%.*]] = insertelement <3 x float> [[TMP4]], float [[TMP2]], i32 1
  14 ; NON-POW2-NEXT:    [[TMP6:%.*]] = call <3 x float> @llvm.fmuladd.v3f32(<3 x float> [[TMP5]], <3 x float> zeroinitializer, <3 x float> zeroinitializer)
  15 ; NON-POW2-NEXT:    store <3 x float> [[TMP6]], ptr [[COLOUR]], align 4
  16 ; NON-POW2-NEXT:    ret void
  17 ;
  18 ; POW2-ONLY-LABEL: @vec3_vectorize_call(
  19 ; POW2-ONLY-NEXT:  entry:
  20 ; POW2-ONLY-NEXT:    [[TMP1:%.*]] = load <2 x float>, ptr [[COLOUR:%.*]], align 4
  21 ; POW2-ONLY-NEXT:    [[TMP2:%.*]] = call <2 x float> @llvm.fmuladd.v2f32(<2 x float> [[TMP1]], <2 x float> zeroinitializer, <2 x float> zeroinitializer)
  22 ; POW2-ONLY-NEXT:    store <2 x float> [[TMP2]], ptr [[COLOUR]], align 4
  23 ; POW2-ONLY-NEXT:    [[ARRAYIDX99_I1:%.*]] = getelementptr float, ptr [[COLOUR]], i64 2
  24 ; POW2-ONLY-NEXT:    [[TMP3:%.*]] = call float @llvm.fmuladd.f32(float [[TMP0:%.*]], float 0.000000e+00, float 0.000000e+00)
  25 ; POW2-ONLY-NEXT:    store float [[TMP3]], ptr [[ARRAYIDX99_I1]], align 4
  26 ; POW2-ONLY-NEXT:    ret void
  27 ;
  28 entry:
  29   %1 = load float, ptr %Colour, align 4
  30   %2 = call float @llvm.fmuladd.f32(float %1, float 0.000000e+00, float 0.000000e+00)
  31   store float %2, ptr %Colour, align 4
  32   %arrayidx91.i = getelementptr float, ptr %Colour, i64 1
  33   %3 = load float, ptr %arrayidx91.i, align 4
  34   %4 = call float @llvm.fmuladd.f32(float %3, float 0.000000e+00, float 0.000000e+00)
  35   store float %4, ptr %arrayidx91.i, align 4
  36   %arrayidx99.i1 = getelementptr float, ptr %Colour, i64 2
  37   %5 = call float @llvm.fmuladd.f32(float %0, float 0.000000e+00, float 0.000000e+00)
  38   store float %5, ptr %arrayidx99.i1, align 4
  39   ret void
  40 }
  41
  42 define void @vec3_fmuladd_64(ptr %Colour, double %0) {
  43 ; CHECK-LABEL: @vec3_fmuladd_64(
  44 ; CHECK-NEXT:  entry:
  45 ; CHECK-NEXT:    [[ARRAYIDX80:%.*]] = getelementptr float, ptr [[COLOUR:%.*]], i64 2
  46 ; CHECK-NEXT:    [[TMP1:%.*]] = insertelement <2 x double> poison, double [[TMP0:%.*]], i32 0
  47 ; CHECK-NEXT:    [[TMP2:%.*]] = shufflevector <2 x double> [[TMP1]], <2 x double> poison, <2 x i32> zeroinitializer
  48 ; CHECK-NEXT:    [[TMP3:%.*]] = call <2 x double> @llvm.fmuladd.v2f64(<2 x double> [[TMP2]], <2 x double> zeroinitializer, <2 x double> zeroinitializer)
  49 ; CHECK-NEXT:    [[TMP4:%.*]] = fptrunc <2 x double> [[TMP3]] to <2 x float>
  50 ; CHECK-NEXT:    store <2 x float> [[TMP4]], ptr [[COLOUR]], align 4
  51 ; CHECK-NEXT:    [[TMP5:%.*]] = call double @llvm.fmuladd.f64(double [[TMP0]], double 0.000000e+00, double 0.000000e+00)
  52 ; CHECK-NEXT:    [[CONV82:%.*]] = fptrunc double [[TMP5]] to float
  53 ; CHECK-NEXT:    store float [[CONV82]], ptr [[ARRAYIDX80]], align 4
  54 ; CHECK-NEXT:    ret void
  55 ;
  56 entry:
  57   %arrayidx72 = getelementptr float, ptr %Colour, i64 1
  58   %arrayidx80 = getelementptr float, ptr %Colour, i64 2
  59   %1 = call double @llvm.fmuladd.f64(double %0, double 0.000000e+00, double 0.000000e+00)
  60   %conv66 = fptrunc double %1 to float
  61   store float %conv66, ptr %Colour, align 4
  62   %2 = call double @llvm.fmuladd.f64(double %0, double 0.000000e+00, double 0.000000e+00)
  63   %conv74 = fptrunc double %2 to float
  64   store float %conv74, ptr %arrayidx72, align 4
  65   %3 = call double @llvm.fmuladd.f64(double %0, double 0.000000e+00, double 0.000000e+00)
  66   %conv82 = fptrunc double %3 to float
  67   store float %conv82, ptr %arrayidx80, align 4
  68   ret void
  69 }
  70
  71 declare float @llvm.fmuladd.f32(float, float, float)
  72
  73 declare double @llvm.fmuladd.f64(double, double, double)