test/CodeGen/X86/vselect.ll

   1 ; RUN: llc -mtriple=x86_64-unknown-unknown -mattr=+sse2,-sse4.1 < %s | FileCheck %s
   2
   3 ; Verify that we don't emit packed vector shifts instructions if the
   4 ; condition used by the vector select is a vector of constants.
   5
   6 define <4 x float> @test1(<4 x float> %a, <4 x float> %b) {
   7 ; CHECK-LABEL: test1:
   8 ; CHECK:       # BB#0:
   9 ; CHECK-NEXT:    shufps {{.*#+}} xmm0 = xmm0[0,2],xmm1[1,3]
  10 ; CHECK-NEXT:    shufps {{.*#+}} xmm0 = xmm0[0,2,1,3]
  11 ; CHECK-NEXT:    retq
  12   %1 = select <4 x i1> <i1 true, i1 false, i1 true, i1 false>, <4 x float> %a, <4 x float> %b
  13   ret <4 x float> %1
  14 }
  15
  16 define <4 x float> @test2(<4 x float> %a, <4 x float> %b) {
  17 ; CHECK-LABEL: test2:
  18 ; CHECK:       # BB#0:
  19 ; CHECK-NEXT:    movsd {{.*#+}} xmm1 = xmm0[0],xmm1[1]
  20 ; CHECK-NEXT:    movapd %xmm1, %xmm0
  21 ; CHECK-NEXT:    retq
  22   %1 = select <4 x i1> <i1 true, i1 true, i1 false, i1 false>, <4 x float> %a, <4 x float> %b
  23   ret <4 x float> %1
  24 }
  25
  26 define <4 x float> @test3(<4 x float> %a, <4 x float> %b) {
  27 ; CHECK-LABEL: test3:
  28 ; CHECK:       # BB#0:
  29 ; CHECK-NEXT:    movsd {{.*#+}} xmm0 = xmm1[0],xmm0[1]
  30 ; CHECK-NEXT:    retq
  31   %1 = select <4 x i1> <i1 false, i1 false, i1 true, i1 true>, <4 x float> %a, <4 x float> %b
  32   ret <4 x float> %1
  33 }
  34
  35 define <4 x float> @test4(<4 x float> %a, <4 x float> %b) {
  36 ; CHECK-LABEL: test4:
  37 ; CHECK:       # BB#0:
  38 ; CHECK-NEXT:    movaps %xmm1, %xmm0
  39 ; CHECK-NEXT:    retq
  40   %1 = select <4 x i1> <i1 false, i1 false, i1 false, i1 false>, <4 x float> %a, <4 x float> %b
  41   ret <4 x float> %1
  42 }
  43
  44 define <4 x float> @test5(<4 x float> %a, <4 x float> %b) {
  45 ; CHECK-LABEL: test5:
  46 ; CHECK:       # BB#0:
  47 ; CHECK-NEXT:    retq
  48   %1 = select <4 x i1> <i1 true, i1 true, i1 true, i1 true>, <4 x float> %a, <4 x float> %b
  49   ret <4 x float> %1
  50 }
  51
  52 define <8 x i16> @test6(<8 x i16> %a, <8 x i16> %b) {
  53 ; CHECK-LABEL: test6:
  54 ; CHECK:       # BB#0:
  55 ; CHECK-NEXT:    retq
  56   %1 = select <8 x i1> <i1 true, i1 false, i1 true, i1 false, i1 true, i1 false, i1 true, i1 false>, <8 x i16> %a, <8 x i16> %a
  57   ret <8 x i16> %1
  58 }
  59
  60 define <8 x i16> @test7(<8 x i16> %a, <8 x i16> %b) {
  61 ; CHECK-LABEL: test7:
  62 ; CHECK:       # BB#0:
  63 ; CHECK-NEXT:    movsd {{.*#+}} xmm1 = xmm0[0],xmm1[1]
  64 ; CHECK-NEXT:    movapd %xmm1, %xmm0
  65 ; CHECK-NEXT:    retq
  66   %1 = select <8 x i1> <i1 true, i1 true, i1 true, i1 true, i1 false, i1 false, i1 false, i1 false>, <8 x i16> %a, <8 x i16> %b
  67   ret <8 x i16> %1
  68 }
  69
  70 define <8 x i16> @test8(<8 x i16> %a, <8 x i16> %b) {
  71 ; CHECK-LABEL: test8:
  72 ; CHECK:       # BB#0:
  73 ; CHECK-NEXT:    movsd {{.*#+}} xmm0 = xmm1[0],xmm0[1]
  74 ; CHECK-NEXT:    retq
  75   %1 = select <8 x i1> <i1 false, i1 false, i1 false, i1 false, i1 true, i1 true, i1 true, i1 true>, <8 x i16> %a, <8 x i16> %b
  76   ret <8 x i16> %1
  77 }
  78
  79 define <8 x i16> @test9(<8 x i16> %a, <8 x i16> %b) {
  80 ; CHECK-LABEL: test9:
  81 ; CHECK:       # BB#0:
  82 ; CHECK-NEXT:    movaps %xmm1, %xmm0
  83 ; CHECK-NEXT:    retq
  84   %1 = select <8 x i1> <i1 false, i1 false, i1 false, i1 false, i1 false, i1 false, i1 false, i1 false>, <8 x i16> %a, <8 x i16> %b
  85   ret <8 x i16> %1
  86 }
  87
  88 define <8 x i16> @test10(<8 x i16> %a, <8 x i16> %b) {
  89 ; CHECK-LABEL: test10:
  90 ; CHECK:       # BB#0:
  91 ; CHECK-NEXT:    retq
  92   %1 = select <8 x i1> <i1 true, i1 true, i1 true, i1 true, i1 true, i1 true, i1 true, i1 true>, <8 x i16> %a, <8 x i16> %b
  93   ret <8 x i16> %1
  94 }
  95
  96 define <8 x i16> @test11(<8 x i16> %a, <8 x i16> %b) {
  97 ; CHECK-LABEL: test11:
  98 ; CHECK:       # BB#0:
  99 ; CHECK-NEXT:    movaps {{.*#+}} xmm2 = [0,65535,65535,0,65535,65535,65535,65535]
 100 ; CHECK-NEXT:    andps %xmm2, %xmm0
 101 ; CHECK-NEXT:    andnps %xmm1, %xmm2
 102 ; CHECK-NEXT:    orps %xmm2, %xmm0
 103 ; CHECK-NEXT:    retq
 104   %1 = select <8 x i1> <i1 false, i1 true, i1 true, i1 false, i1 undef, i1 true, i1 true, i1 undef>, <8 x i16> %a, <8 x i16> %b
 105   ret <8 x i16> %1
 106 }
 107
 108 define <8 x i16> @test12(<8 x i16> %a, <8 x i16> %b) {
 109 ; CHECK-LABEL: test12:
 110 ; CHECK:       # BB#0:
 111 ; CHECK-NEXT:    movaps %xmm1, %xmm0
 112 ; CHECK-NEXT:    retq
 113   %1 = select <8 x i1> <i1 false, i1 false, i1 undef, i1 false, i1 false, i1 false, i1 false, i1 undef>, <8 x i16> %a, <8 x i16> %b
 114   ret <8 x i16> %1
 115 }
 116
 117 define <8 x i16> @test13(<8 x i16> %a, <8 x i16> %b) {
 118 ; CHECK-LABEL: test13:
 119 ; CHECK:       # BB#0:
 120 ; CHECK-NEXT:    movaps %xmm1, %xmm0
 121 ; CHECK-NEXT:    retq
 122   %1 = select <8 x i1> <i1 undef, i1 undef, i1 undef, i1 undef, i1 undef, i1 undef, i1 undef, i1 undef>, <8 x i16> %a, <8 x i16> %b
 123   ret <8 x i16> %1
 124 }
 125
 126 ; Fold (vselect (build_vector AllOnes), N1, N2) -> N1
 127 define <4 x float> @test14(<4 x float> %a, <4 x float> %b) {
 128 ; CHECK-LABEL: test14:
 129 ; CHECK:       # BB#0:
 130 ; CHECK-NEXT:    retq
 131   %1 = select <4 x i1> <i1 true, i1 undef, i1 true, i1 undef>, <4 x float> %a, <4 x float> %b
 132   ret <4 x float> %1
 133 }
 134
 135 define <8 x i16> @test15(<8 x i16> %a, <8 x i16> %b) {
 136 ; CHECK-LABEL: test15:
 137 ; CHECK:       # BB#0:
 138 ; CHECK-NEXT:    retq
 139   %1 = select <8 x i1> <i1 true, i1 true, i1 true, i1 undef, i1 undef, i1 true, i1 true, i1 undef>, <8 x i16> %a, <8 x i16> %b
 140   ret <8 x i16> %1
 141 }
 142
 143 ; Fold (vselect (build_vector AllZeros), N1, N2) -> N2
 144 define <4 x float> @test16(<4 x float> %a, <4 x float> %b) {
 145 ; CHECK-LABEL: test16:
 146 ; CHECK:       # BB#0:
 147 ; CHECK-NEXT:    movaps %xmm1, %xmm0
 148 ; CHECK-NEXT:    retq
 149   %1 = select <4 x i1> <i1 false, i1 undef, i1 false, i1 undef>, <4 x float> %a, <4 x float> %b
 150   ret <4 x float> %1
 151 }
 152
 153 define <8 x i16> @test17(<8 x i16> %a, <8 x i16> %b) {
 154 ; CHECK-LABEL: test17:
 155 ; CHECK:       # BB#0:
 156 ; CHECK-NEXT:    movaps %xmm1, %xmm0
 157 ; CHECK-NEXT:    retq
 158   %1 = select <8 x i1> <i1 false, i1 false, i1 false, i1 undef, i1 undef, i1 false, i1 false, i1 undef>, <8 x i16> %a, <8 x i16> %b
 159   ret <8 x i16> %1
 160 }
 161
 162 define <4 x float> @test18(<4 x float> %a, <4 x float> %b) {
 163 ; CHECK-LABEL: test18:
 164 ; CHECK:       # BB#0:
 165 ; CHECK-NEXT:    movss {{.*#+}} xmm0 = xmm1[0],xmm0[1,2,3]
 166 ; CHECK-NEXT:    retq
 167   %1 = select <4 x i1> <i1 false, i1 true, i1 true, i1 true>, <4 x float> %a, <4 x float> %b
 168   ret <4 x float> %1
 169 }
 170
 171 define <4 x i32> @test19(<4 x i32> %a, <4 x i32> %b) {
 172 ; CHECK-LABEL: test19:
 173 ; CHECK:       # BB#0:
 174 ; CHECK-NEXT:    movss {{.*#+}} xmm0 = xmm1[0],xmm0[1,2,3]
 175 ; CHECK-NEXT:    retq
 176   %1 = select <4 x i1> <i1 false, i1 true, i1 true, i1 true>, <4 x i32> %a, <4 x i32> %b
 177   ret <4 x i32> %1
 178 }
 179
 180 define <2 x double> @test20(<2 x double> %a, <2 x double> %b) {
 181 ; CHECK-LABEL: test20:
 182 ; CHECK:       # BB#0:
 183 ; CHECK-NEXT:    movsd {{.*#+}} xmm0 = xmm1[0],xmm0[1]
 184 ; CHECK-NEXT:    retq
 185   %1 = select <2 x i1> <i1 false, i1 true>, <2 x double> %a, <2 x double> %b
 186   ret <2 x double> %1
 187 }
 188
 189 define <2 x i64> @test21(<2 x i64> %a, <2 x i64> %b) {
 190 ; CHECK-LABEL: test21:
 191 ; CHECK:       # BB#0:
 192 ; CHECK-NEXT:    movsd {{.*#+}} xmm0 = xmm1[0],xmm0[1]
 193 ; CHECK-NEXT:    retq
 194   %1 = select <2 x i1> <i1 false, i1 true>, <2 x i64> %a, <2 x i64> %b
 195   ret <2 x i64> %1
 196 }
 197
 198 define <4 x float> @test22(<4 x float> %a, <4 x float> %b) {
 199 ; CHECK-LABEL: test22:
 200 ; CHECK:       # BB#0:
 201 ; CHECK-NEXT:    movss {{.*#+}} xmm1 = xmm0[0],xmm1[1,2,3]
 202 ; CHECK-NEXT:    movaps %xmm1, %xmm0
 203 ; CHECK-NEXT:    retq
 204   %1 = select <4 x i1> <i1 true, i1 false, i1 false, i1 false>, <4 x float> %a, <4 x float> %b
 205   ret <4 x float> %1
 206 }
 207
 208 define <4 x i32> @test23(<4 x i32> %a, <4 x i32> %b) {
 209 ; CHECK-LABEL: test23:
 210 ; CHECK:       # BB#0:
 211 ; CHECK-NEXT:    movss {{.*#+}} xmm1 = xmm0[0],xmm1[1,2,3]
 212 ; CHECK-NEXT:    movaps %xmm1, %xmm0
 213 ; CHECK-NEXT:    retq
 214   %1 = select <4 x i1> <i1 true, i1 false, i1 false, i1 false>, <4 x i32> %a, <4 x i32> %b
 215   ret <4 x i32> %1
 216 }
 217
 218 define <2 x double> @test24(<2 x double> %a, <2 x double> %b) {
 219 ; CHECK-LABEL: test24:
 220 ; CHECK:       # BB#0:
 221 ; CHECK-NEXT:    movsd {{.*#+}} xmm1 = xmm0[0],xmm1[1]
 222 ; CHECK-NEXT:    movapd %xmm1, %xmm0
 223 ; CHECK-NEXT:    retq
 224   %1 = select <2 x i1> <i1 true, i1 false>, <2 x double> %a, <2 x double> %b
 225   ret <2 x double> %1
 226 }
 227
 228 define <2 x i64> @test25(<2 x i64> %a, <2 x i64> %b) {
 229 ; CHECK-LABEL: test25:
 230 ; CHECK:       # BB#0:
 231 ; CHECK-NEXT:    movsd {{.*#+}} xmm1 = xmm0[0],xmm1[1]
 232 ; CHECK-NEXT:    movapd %xmm1, %xmm0
 233 ; CHECK-NEXT:    retq
 234   %1 = select <2 x i1> <i1 true, i1 false>, <2 x i64> %a, <2 x i64> %b
 235   ret <2 x i64> %1
 236 }
 237
 238 define <4 x float> @select_of_shuffles_0(<2 x float> %a0, <2 x float> %b0, <2 x float> %a1, <2 x float> %b1) {
 239 ; CHECK-LABEL: select_of_shuffles_0:
 240 ; CHECK:       # BB#0:
 241 ; CHECK-NEXT:    unpcklpd {{.*#+}} xmm0 = xmm0[0],xmm2[0]
 242 ; CHECK-NEXT:    unpcklpd {{.*#+}} xmm1 = xmm1[0],xmm3[0]
 243 ; CHECK-NEXT:    subps %xmm1, %xmm0
 244 ; CHECK-NEXT:    retq
 245   %1 = shufflevector <2 x float> %a0, <2 x float> undef, <4 x i32> <i32 0, i32 1, i32 undef, i32 undef>
 246   %2 = shufflevector <2 x float> %a1, <2 x float> undef, <4 x i32> <i32 undef, i32 undef, i32 0, i32 1>
 247   %3 = select <4 x i1> <i1 false, i1 false, i1 true, i1 true>, <4 x float> %2, <4 x float> %1
 248   %4 = shufflevector <2 x float> %b0, <2 x float> undef, <4 x i32> <i32 0, i32 1, i32 undef, i32 undef>
 249   %5 = shufflevector <2 x float> %b1, <2 x float> undef, <4 x i32> <i32 undef, i32 undef, i32 0, i32 1>
 250   %6 = select <4 x i1> <i1 false, i1 false, i1 true, i1 true>, <4 x float> %5, <4 x float> %4
 251   %7 = fsub <4 x float> %3, %6
 252   ret <4 x float> %7
 253 }
 254
 255 ; PR20677
 256 define <16 x double> @select_illegal(<16 x double> %a, <16 x double> %b) {
 257 ; CHECK-LABEL: select_illegal:
 258 ; CHECK:       # BB#0:
 259 ; CHECK-NEXT:    movaps {{[0-9]+}}(%rsp), %xmm4
 260 ; CHECK-NEXT:    movaps {{[0-9]+}}(%rsp), %xmm5
 261 ; CHECK-NEXT:    movaps {{[0-9]+}}(%rsp), %xmm6
 262 ; CHECK-NEXT:    movaps {{[0-9]+}}(%rsp), %xmm7
 263 ; CHECK-NEXT:    movaps %xmm7, 112(%rdi)
 264 ; CHECK-NEXT:    movaps %xmm6, 96(%rdi)
 265 ; CHECK-NEXT:    movaps %xmm5, 80(%rdi)
 266 ; CHECK-NEXT:    movaps %xmm4, 64(%rdi)
 267 ; CHECK-NEXT:    movaps %xmm3, 48(%rdi)
 268 ; CHECK-NEXT:    movaps %xmm2, 32(%rdi)
 269 ; CHECK-NEXT:    movaps %xmm1, 16(%rdi)
 270 ; CHECK-NEXT:    movaps %xmm0, (%rdi)
 271 ; CHECK-NEXT:    movq %rdi, %rax
 272 ; CHECK-NEXT:    retq
 273   %sel = select <16 x i1> <i1 true, i1 true, i1 true, i1 true, i1 true, i1 true, i1 true, i1 true, i1 false, i1 false, i1 false, i1 false, i1 false, i1 false, i1 false, i1 false>, <16 x double> %a, <16 x double> %b
 274   ret <16 x double> %sel
 275 }