test/Transforms/BBVectorize/simple.ll

   1 target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128"
   2 ; RUN: opt < %s -bb-vectorize -bb-vectorize-req-chain-depth=3 -instcombine -gvn -S | FileCheck %s
   3 ; RUN: opt < %s -mtriple=x86_64-unknown-linux-gnu -bb-vectorize -bb-vectorize-req-chain-depth=3 -instcombine -gvn -S | FileCheck %s -check-prefix=CHECK-TI
   4
   5 ; Basic depth-3 chain
   6 define double @test1(double %A1, double %A2, double %B1, double %B2) {
   7 ; CHECK: @test1
   8 ; CHECK: %X1.v.i1.1 = insertelement <2 x double> undef, double %B1, i32 0
   9 ; CHECK: %X1.v.i0.1 = insertelement <2 x double> undef, double %A1, i32 0
  10 ; CHECK: %X1.v.i1.2 = insertelement <2 x double> %X1.v.i1.1, double %B2, i32 1
  11 ; CHECK: %X1.v.i0.2 = insertelement <2 x double> %X1.v.i0.1, double %A2, i32 1
  12         %X1 = fsub double %A1, %B1
  13         %X2 = fsub double %A2, %B2
  14 ; CHECK: %X1 = fsub <2 x double> %X1.v.i0.2, %X1.v.i1.2
  15         %Y1 = fmul double %X1, %A1
  16         %Y2 = fmul double %X2, %A2
  17 ; CHECK: %Y1 = fmul <2 x double> %X1, %X1.v.i0.2
  18         %Z1 = fadd double %Y1, %B1
  19         %Z2 = fadd double %Y2, %B2
  20 ; CHECK: %Z1 = fadd <2 x double> %Y1, %X1.v.i1.2
  21         %R  = fmul double %Z1, %Z2
  22 ; CHECK: %Z1.v.r1 = extractelement <2 x double> %Z1, i32 0
  23 ; CHECK: %Z1.v.r2 = extractelement <2 x double> %Z1, i32 1
  24 ; CHECK: %R = fmul double %Z1.v.r1, %Z1.v.r2
  25         ret double %R
  26 ; CHECK: ret double %R
  27 ; CHECK-TI: @test1
  28 ; CHECK-TI: fsub <2 x double>
  29 ; CHECK-TI: ret double
  30 }
  31
  32 ; Basic depth-3 chain (last pair permuted)
  33 define double @test2(double %A1, double %A2, double %B1, double %B2) {
  34 ; CHECK: @test2
  35 ; CHECK: %X1.v.i1.1 = insertelement <2 x double> undef, double %B1, i32 0
  36 ; CHECK: %X1.v.i0.1 = insertelement <2 x double> undef, double %A1, i32 0
  37 ; CHECK: %X1.v.i1.2 = insertelement <2 x double> %X1.v.i1.1, double %B2, i32 1
  38 ; CHECK: %X1.v.i0.2 = insertelement <2 x double> %X1.v.i0.1, double %A2, i32 1
  39         %X1 = fsub double %A1, %B1
  40         %X2 = fsub double %A2, %B2
  41 ; CHECK: %X1 = fsub <2 x double> %X1.v.i0.2, %X1.v.i1.2
  42         %Y1 = fmul double %X1, %A1
  43         %Y2 = fmul double %X2, %A2
  44 ; CHECK: %Y1 = fmul <2 x double> %X1, %X1.v.i0.2
  45         %Z1 = fadd double %Y2, %B1
  46         %Z2 = fadd double %Y1, %B2
  47 ; CHECK: %Z1.v.i0 = shufflevector <2 x double> %Y1, <2 x double> undef, <2 x i32> <i32 1, i32 0>
  48 ; CHECK: %Z1 = fadd <2 x double> %Z1.v.i0, %X1.v.i1.2
  49         %R  = fmul double %Z1, %Z2
  50 ; CHECK: %Z1.v.r1 = extractelement <2 x double> %Z1, i32 0
  51 ; CHECK: %Z1.v.r2 = extractelement <2 x double> %Z1, i32 1
  52 ; CHECK: %R = fmul double %Z1.v.r1, %Z1.v.r2
  53         ret double %R
  54 ; CHECK: ret double %R
  55 }
  56
  57 ; Basic depth-3 chain (last pair first splat)
  58 define double @test3(double %A1, double %A2, double %B1, double %B2) {
  59 ; CHECK: @test3
  60 ; CHECK: %X1.v.i1.1 = insertelement <2 x double> undef, double %B1, i32 0
  61 ; CHECK: %X1.v.i0.1 = insertelement <2 x double> undef, double %A1, i32 0
  62 ; CHECK: %X1.v.i1.2 = insertelement <2 x double> %X1.v.i1.1, double %B2, i32 1
  63 ; CHECK: %X1.v.i0.2 = insertelement <2 x double> %X1.v.i0.1, double %A2, i32 1
  64         %X1 = fsub double %A1, %B1
  65         %X2 = fsub double %A2, %B2
  66 ; CHECK: %X1 = fsub <2 x double> %X1.v.i0.2, %X1.v.i1.2
  67         %Y1 = fmul double %X1, %A1
  68         %Y2 = fmul double %X2, %A2
  69 ; CHECK: %Y1 = fmul <2 x double> %X1, %X1.v.i0.2
  70         %Z1 = fadd double %Y2, %B1
  71         %Z2 = fadd double %Y2, %B2
  72 ; CHECK: %Z1.v.i0 = shufflevector <2 x double> %Y1, <2 x double> undef, <2 x i32> <i32 1, i32 1>
  73 ; CHECK: %Z1 = fadd <2 x double> %Z1.v.i0, %X1.v.i1.2
  74         %R  = fmul double %Z1, %Z2
  75 ; CHECK: %Z1.v.r1 = extractelement <2 x double> %Z1, i32 0
  76 ; CHECK: %Z1.v.r2 = extractelement <2 x double> %Z1, i32 1
  77 ; CHECK: %R = fmul double %Z1.v.r1, %Z1.v.r2
  78         ret double %R
  79 ; CHECK: ret double %R
  80 }
  81
  82 ; Basic depth-3 chain (last pair second splat)
  83 define double @test4(double %A1, double %A2, double %B1, double %B2) {
  84 ; CHECK: @test4
  85 ; CHECK: %X1.v.i1.1 = insertelement <2 x double> undef, double %B1, i32 0
  86 ; CHECK: %X1.v.i0.1 = insertelement <2 x double> undef, double %A1, i32 0
  87 ; CHECK: %X1.v.i1.2 = insertelement <2 x double> %X1.v.i1.1, double %B2, i32 1
  88 ; CHECK: %X1.v.i0.2 = insertelement <2 x double> %X1.v.i0.1, double %A2, i32 1
  89         %X1 = fsub double %A1, %B1
  90         %X2 = fsub double %A2, %B2
  91 ; CHECK: %X1 = fsub <2 x double> %X1.v.i0.2, %X1.v.i1.2
  92         %Y1 = fmul double %X1, %A1
  93         %Y2 = fmul double %X2, %A2
  94 ; CHECK: %Y1 = fmul <2 x double> %X1, %X1.v.i0.2
  95         %Z1 = fadd double %Y1, %B1
  96         %Z2 = fadd double %Y1, %B2
  97 ; CHECK: %Z1.v.i0 = shufflevector <2 x double> %Y1, <2 x double> undef, <2 x i32> zeroinitializer
  98 ; CHECK: %Z1 = fadd <2 x double> %Z1.v.i0, %X1.v.i1.2
  99         %R  = fmul double %Z1, %Z2
 100 ; CHECK: %Z1.v.r1 = extractelement <2 x double> %Z1, i32 0
 101 ; CHECK: %Z1.v.r2 = extractelement <2 x double> %Z1, i32 1
 102 ; CHECK: %R = fmul double %Z1.v.r1, %Z1.v.r2
 103         ret double %R
 104 ; CHECK: ret double %R
 105 }
 106
 107 ; Basic depth-3 chain
 108 define <2 x float> @test5(<2 x float> %A1, <2 x float> %A2, <2 x float> %B1, <2 x float> %B2) {
 109 ; CHECK: @test5
 110 ; CHECK: %X1.v.i1 = shufflevector <2 x float> %B1, <2 x float> %B2, <4 x i32> <i32 0, i32 1, i32 2, i32 3>
 111 ; CHECK: %X1.v.i0 = shufflevector <2 x float> %A1, <2 x float> %A2, <4 x i32> <i32 0, i32 1, i32 2, i32 3>
 112         %X1 = fsub <2 x float> %A1, %B1
 113         %X2 = fsub <2 x float> %A2, %B2
 114 ; CHECK: %X1 = fsub <4 x float> %X1.v.i0, %X1.v.i1
 115         %Y1 = fmul <2 x float> %X1, %A1
 116         %Y2 = fmul <2 x float> %X2, %A2
 117 ; CHECK: %Y1 = fmul <4 x float> %X1, %X1.v.i0
 118         %Z1 = fadd <2 x float> %Y1, %B1
 119         %Z2 = fadd <2 x float> %Y2, %B2
 120 ; CHECK: %Z1 = fadd <4 x float> %Y1, %X1.v.i1
 121         %R  = fmul <2 x float> %Z1, %Z2
 122 ; CHECK: %Z1.v.r1 = shufflevector <4 x float> %Z1, <4 x float> undef, <2 x i32> <i32 0, i32 1>
 123 ; CHECK: %Z1.v.r2 = shufflevector <4 x float> %Z1, <4 x float> undef, <2 x i32> <i32 2, i32 3>
 124 ; CHECK: %R = fmul <2 x float> %Z1.v.r1, %Z1.v.r2
 125         ret <2 x float> %R
 126 ; CHECK: ret <2 x float> %R
 127 }
 128
 129 ; Basic chain with shuffles
 130 define <8 x i8> @test6(<8 x i8> %A1, <8 x i8> %A2, <8 x i8> %B1, <8 x i8> %B2) {
 131 ; CHECK: @test6
 132 ; CHECK: %X1.v.i1 = shufflevector <8 x i8> %B1, <8 x i8> %B2, <16 x i32> <i32 0, i32 1, i32 2, i32 3, i32 4, i32 5, i32 6, i32 7, i32 8, i32 9, i32 10, i32 11, i32 12, i32 13, i32 14, i32 15>
 133 ; CHECK: %X1.v.i0 = shufflevector <8 x i8> %A1, <8 x i8> %A2, <16 x i32> <i32 0, i32 1, i32 2, i32 3, i32 4, i32 5, i32 6, i32 7, i32 8, i32 9, i32 10, i32 11, i32 12, i32 13, i32 14, i32 15>
 134         %X1 = sub <8 x i8> %A1, %B1
 135         %X2 = sub <8 x i8> %A2, %B2
 136 ; CHECK: %X1 = sub <16 x i8> %X1.v.i0, %X1.v.i1
 137         %Y1 = mul <8 x i8> %X1, %A1
 138         %Y2 = mul <8 x i8> %X2, %A2
 139 ; CHECK: %Y1 = mul <16 x i8> %X1, %X1.v.i0
 140         %Z1 = add <8 x i8> %Y1, %B1
 141         %Z2 = add <8 x i8> %Y2, %B2
 142 ; CHECK: %Z1 = add <16 x i8> %Y1, %X1.v.i1
 143         %Q1 = shufflevector <8 x i8> %Z1, <8 x i8> %Z2, <8 x i32> <i32 15, i32 8, i32 6, i32 1, i32 13, i32 10, i32 4, i32 3>
 144         %Q2 = shufflevector <8 x i8> %Z2, <8 x i8> %Z2, <8 x i32> <i32 6, i32 7, i32 0, i32 1, i32 2, i32 4, i32 4, i32 1>
 145 ; CHECK: %Q1.v.i1 = shufflevector <16 x i8> %Z1, <16 x i8> undef, <16 x i32> <i32 8, i32 undef, i32 10, i32 undef, i32 undef, i32 13, i32 undef, i32 15, i32 undef, i32 undef, i32 undef, i32 undef, i32 undef, i32 undef, i32 undef, i32 undef>
 146 ; CHECK: %Q1 = shufflevector <16 x i8> %Z1, <16 x i8> %Q1.v.i1, <16 x i32> <i32 23, i32 16, i32 6, i32 1, i32 21, i32 18, i32 4, i32 3, i32 14, i32 15, i32 8, i32 9, i32 10, i32 12, i32 12, i32 9>
 147         %R  = mul <8 x i8> %Q1, %Q2
 148 ; CHECK: %Q1.v.r1 = shufflevector <16 x i8> %Q1, <16 x i8> undef, <8 x i32> <i32 0, i32 1, i32 2, i32 3, i32 4, i32 5, i32 6, i32 7>
 149 ; CHECK: %Q1.v.r2 = shufflevector <16 x i8> %Q1, <16 x i8> undef, <8 x i32> <i32 8, i32 9, i32 10, i32 11, i32 12, i32 13, i32 14, i32 15>
 150 ; CHECK: %R = mul <8 x i8> %Q1.v.r1, %Q1.v.r2
 151         ret <8 x i8> %R
 152 ; CHECK: ret <8 x i8> %R
 153 ; CHECK-TI: @test6
 154 ; CHECK-TI-NOT: sub <16 x i8>
 155 ; CHECK-TI: ret <8 x i8>
 156 }
 157
 158