test/CodeGen/X86/avx512-vbroadcast.ll

   1 ; RUN: llc < %s -mtriple=x86_64-apple-darwin -mcpu=knl --show-mc-encoding| FileCheck %s
   2
   3 ;CHECK-LABEL: _inreg16xi32:
   4 ;CHECK: vpbroadcastd {{.*}}, %zmm
   5 ;CHECK: ret
   6 define   <16 x i32> @_inreg16xi32(i32 %a) {
   7   %b = insertelement <16 x i32> undef, i32 %a, i32 0
   8   %c = shufflevector <16 x i32> %b, <16 x i32> undef, <16 x i32> zeroinitializer
   9   ret <16 x i32> %c
  10 }
  11
  12 ;CHECK-LABEL: _inreg8xi64:
  13 ;CHECK: vpbroadcastq {{.*}}, %zmm
  14 ;CHECK: ret
  15 define   <8 x i64> @_inreg8xi64(i64 %a) {
  16   %b = insertelement <8 x i64> undef, i64 %a, i32 0
  17   %c = shufflevector <8 x i64> %b, <8 x i64> undef, <8 x i32> zeroinitializer
  18   ret <8 x i64> %c
  19 }
  20
  21 ;CHECK-LABEL: _inreg16xfloat:
  22 ;CHECK: vbroadcastss {{.*}}, %zmm
  23 ;CHECK: ret
  24 define   <16 x float> @_inreg16xfloat(float %a) {
  25   %b = insertelement <16 x float> undef, float %a, i32 0
  26   %c = shufflevector <16 x float> %b, <16 x float> undef, <16 x i32> zeroinitializer
  27   ret <16 x float> %c
  28 }
  29
  30 ;CHECK-LABEL: _inreg8xdouble:
  31 ;CHECK: vbroadcastsd {{.*}}, %zmm
  32 ;CHECK: ret
  33 define   <8 x double> @_inreg8xdouble(double %a) {
  34   %b = insertelement <8 x double> undef, double %a, i32 0
  35   %c = shufflevector <8 x double> %b, <8 x double> undef, <8 x i32> zeroinitializer
  36   ret <8 x double> %c
  37 }
  38
  39 ;CHECK-LABEL: _xmm16xi32
  40 ;CHECK: vpbroadcastd
  41 ;CHECK: ret
  42 define   <16 x i32> @_xmm16xi32(<16 x i32> %a) {
  43   %b = shufflevector <16 x i32> %a, <16 x i32> undef, <16 x i32> zeroinitializer
  44   ret <16 x i32> %b
  45 }
  46
  47 ;CHECK-LABEL: _xmm16xfloat
  48 ;CHECK: vbroadcastss {{.*}}## encoding: [0x62
  49 ;CHECK: ret
  50 define   <16 x float> @_xmm16xfloat(<16 x float> %a) {
  51   %b = shufflevector <16 x float> %a, <16 x float> undef, <16 x i32> zeroinitializer
  52   ret <16 x float> %b
  53 }
  54
  55 define <16 x i32> @test_vbroadcast() {
  56 ; CHECK-LABEL: _test_vbroadcast:
  57 ; CHECK: vpbroadcastd
  58 entry:
  59   %0 = sext <16 x i1> zeroinitializer to <16 x i32>
  60   %1 = fcmp uno <16 x float> undef, zeroinitializer
  61   %2 = sext <16 x i1> %1 to <16 x i32>
  62   %3 = select <16 x i1> %1, <16 x i32> %0, <16 x i32> %2
  63   ret <16 x i32> %3
  64 }
  65
  66 ; We implement the set1 intrinsics with vector initializers.  Verify that the
  67 ; IR generated will produce broadcasts at the end.
  68 define <8 x double> @test_set1_pd(double %d) #2 {
  69 ; CHECK-LABEL: _test_set1_pd:
  70 ; CHECK: vbroadcastsd
  71 entry:
  72   %vecinit.i = insertelement <8 x double> undef, double %d, i32 0
  73   %vecinit1.i = insertelement <8 x double> %vecinit.i, double %d, i32 1
  74   %vecinit2.i = insertelement <8 x double> %vecinit1.i, double %d, i32 2
  75   %vecinit3.i = insertelement <8 x double> %vecinit2.i, double %d, i32 3
  76   %vecinit4.i = insertelement <8 x double> %vecinit3.i, double %d, i32 4
  77   %vecinit5.i = insertelement <8 x double> %vecinit4.i, double %d, i32 5
  78   %vecinit6.i = insertelement <8 x double> %vecinit5.i, double %d, i32 6
  79   %vecinit7.i = insertelement <8 x double> %vecinit6.i, double %d, i32 7
  80   ret <8 x double> %vecinit7.i
  81 }
  82
  83 define <8 x i64> @test_set1_epi64(i64 %d) #2 {
  84 ; CHECK-LABEL: _test_set1_epi64:
  85 ; CHECK: vpbroadcastq
  86 entry:
  87   %vecinit.i = insertelement <8 x i64> undef, i64 %d, i32 0
  88   %vecinit1.i = insertelement <8 x i64> %vecinit.i, i64 %d, i32 1
  89   %vecinit2.i = insertelement <8 x i64> %vecinit1.i, i64 %d, i32 2
  90   %vecinit3.i = insertelement <8 x i64> %vecinit2.i, i64 %d, i32 3
  91   %vecinit4.i = insertelement <8 x i64> %vecinit3.i, i64 %d, i32 4
  92   %vecinit5.i = insertelement <8 x i64> %vecinit4.i, i64 %d, i32 5
  93   %vecinit6.i = insertelement <8 x i64> %vecinit5.i, i64 %d, i32 6
  94   %vecinit7.i = insertelement <8 x i64> %vecinit6.i, i64 %d, i32 7
  95   ret <8 x i64> %vecinit7.i
  96 }
  97
  98 define <16 x float> @test_set1_ps(float %f) #2 {
  99 ; CHECK-LABEL: _test_set1_ps:
 100 ; CHECK: vbroadcastss
 101 entry:
 102   %vecinit.i = insertelement <16 x float> undef, float %f, i32 0
 103   %vecinit1.i = insertelement <16 x float> %vecinit.i, float %f, i32 1
 104   %vecinit2.i = insertelement <16 x float> %vecinit1.i, float %f, i32 2
 105   %vecinit3.i = insertelement <16 x float> %vecinit2.i, float %f, i32 3
 106   %vecinit4.i = insertelement <16 x float> %vecinit3.i, float %f, i32 4
 107   %vecinit5.i = insertelement <16 x float> %vecinit4.i, float %f, i32 5
 108   %vecinit6.i = insertelement <16 x float> %vecinit5.i, float %f, i32 6
 109   %vecinit7.i = insertelement <16 x float> %vecinit6.i, float %f, i32 7
 110   %vecinit8.i = insertelement <16 x float> %vecinit7.i, float %f, i32 8
 111   %vecinit9.i = insertelement <16 x float> %vecinit8.i, float %f, i32 9
 112   %vecinit10.i = insertelement <16 x float> %vecinit9.i, float %f, i32 10
 113   %vecinit11.i = insertelement <16 x float> %vecinit10.i, float %f, i32 11
 114   %vecinit12.i = insertelement <16 x float> %vecinit11.i, float %f, i32 12
 115   %vecinit13.i = insertelement <16 x float> %vecinit12.i, float %f, i32 13
 116   %vecinit14.i = insertelement <16 x float> %vecinit13.i, float %f, i32 14
 117   %vecinit15.i = insertelement <16 x float> %vecinit14.i, float %f, i32 15
 118   ret <16 x float> %vecinit15.i
 119 }
 120
 121 define <16 x i32> @test_set1_epi32(i32 %f) #2 {
 122 ; CHECK-LABEL: _test_set1_epi32:
 123 ; CHECK: vpbroadcastd
 124 entry:
 125   %vecinit.i = insertelement <16 x i32> undef, i32 %f, i32 0
 126   %vecinit1.i = insertelement <16 x i32> %vecinit.i, i32 %f, i32 1
 127   %vecinit2.i = insertelement <16 x i32> %vecinit1.i, i32 %f, i32 2
 128   %vecinit3.i = insertelement <16 x i32> %vecinit2.i, i32 %f, i32 3
 129   %vecinit4.i = insertelement <16 x i32> %vecinit3.i, i32 %f, i32 4
 130   %vecinit5.i = insertelement <16 x i32> %vecinit4.i, i32 %f, i32 5
 131   %vecinit6.i = insertelement <16 x i32> %vecinit5.i, i32 %f, i32 6
 132   %vecinit7.i = insertelement <16 x i32> %vecinit6.i, i32 %f, i32 7
 133   %vecinit8.i = insertelement <16 x i32> %vecinit7.i, i32 %f, i32 8
 134   %vecinit9.i = insertelement <16 x i32> %vecinit8.i, i32 %f, i32 9
 135   %vecinit10.i = insertelement <16 x i32> %vecinit9.i, i32 %f, i32 10
 136   %vecinit11.i = insertelement <16 x i32> %vecinit10.i, i32 %f, i32 11
 137   %vecinit12.i = insertelement <16 x i32> %vecinit11.i, i32 %f, i32 12
 138   %vecinit13.i = insertelement <16 x i32> %vecinit12.i, i32 %f, i32 13
 139   %vecinit14.i = insertelement <16 x i32> %vecinit13.i, i32 %f, i32 14
 140   %vecinit15.i = insertelement <16 x i32> %vecinit14.i, i32 %f, i32 15
 141   ret <16 x i32> %vecinit15.i
 142 }
 143
 144 ; We implement the scalar broadcast intrinsics with vector initializers.
 145 ; Verify that the IR generated will produce the broadcast at the end.
 146 define <8 x double> @test_mm512_broadcastsd_pd(<2 x double> %a) {
 147 ; CHECK-LABEL: _test_mm512_broadcastsd_pd:
 148 ; CHECK: vbroadcastsd %xmm0, %zmm0
 149 entry:
 150   %0 = extractelement <2 x double> %a, i32 0
 151   %vecinit.i = insertelement <8 x double> undef, double %0, i32 0
 152   %vecinit1.i = insertelement <8 x double> %vecinit.i, double %0, i32 1
 153   %vecinit2.i = insertelement <8 x double> %vecinit1.i, double %0, i32 2
 154   %vecinit3.i = insertelement <8 x double> %vecinit2.i, double %0, i32 3
 155   %vecinit4.i = insertelement <8 x double> %vecinit3.i, double %0, i32 4
 156   %vecinit5.i = insertelement <8 x double> %vecinit4.i, double %0, i32 5
 157   %vecinit6.i = insertelement <8 x double> %vecinit5.i, double %0, i32 6
 158   %vecinit7.i = insertelement <8 x double> %vecinit6.i, double %0, i32 7
 159   ret <8 x double> %vecinit7.i
 160 }