test/CodeGen/X86/vselect-avx.ll

   1 ; RUN: llc %s -o - -mattr=+avx | FileCheck %s
   2 target datalayout = "e-m:o-i64:64-f80:128-n8:16:32:64-S128"
   3 target triple = "x86_64-apple-macosx"
   4
   5 ; For this test we used to optimize the <i1 true, i1 false, i1 false, i1 true>
   6 ; mask into <i32 2147483648, i32 0, i32 0, i32 2147483648> because we thought
   7 ; we would lower that into a blend where only the high bit is relevant.
   8 ; However, since the whole mask is constant, this is simplified incorrectly
   9 ; by the generic code, because it was expecting -1 in place of 2147483648.
  10 ;
  11 ; The problem does not occur without AVX, because vselect of v4i32 is not legal
  12 ; nor custom.
  13 ;
  14 ; <rdar://problem/18675020>
  15
  16 ; CHECK-LABEL: test:
  17 ; CHECK: vmovdqa {{.*#+}} xmm0 = [65535,0,0,65535]
  18 ; CHECK: vmovdqa {{.*#+}} xmm2 = [65533,124,125,14807]
  19 ; CHECK: ret
  20 define void @test(<4 x i16>* %a, <4 x i16>* %b) {
  21 body:
  22   %predphi = select <4 x i1> <i1 true, i1 false, i1 false, i1 true>, <4 x i16> <i16 -3, i16 545, i16 4385, i16 14807>, <4 x i16> <i16 123, i16 124, i16 125, i16 127>
  23   %predphi42 = select <4 x i1> <i1 true, i1 false, i1 false, i1 true>, <4 x i16> <i16 -1, i16 -1, i16 -1, i16 -1>, <4 x i16> zeroinitializer
  24   store <4 x i16> %predphi, <4 x i16>* %a, align 8
  25   store <4 x i16> %predphi42, <4 x i16>* %b, align 8
  26   ret void
  27 }
  28
  29 ; Improve code coverage.
  30 ;
  31 ; When shrinking the condition used into the select to match a blend, this
  32 ; test case exercises the path where the modified node is not the root
  33 ; of the condition.
  34 ;
  35 ; CHECK-LABEL: test2:
  36 ; CHECK:        vpslld  $31, %xmm0, %xmm0
  37 ; CHECK-NEXT:   vpmovsxdq       %xmm0, %xmm1
  38 ; CHECK-NEXT:   vpshufd $78, %xmm0, %xmm0       ## xmm0 = xmm0[2,3,0,1]
  39 ; CHECK-NEXT:   vpmovsxdq       %xmm0, %xmm0
  40 ; CHECK-NEXT:   vinsertf128     $1, %xmm0, %ymm1, [[MASK:%ymm[0-9]+]]
  41 ; CHECK: vblendvpd      [[MASK]]
  42 ; CHECK: retq
  43 define void @test2(double** %call1559, i64 %indvars.iv4198, <4 x i1> %tmp1895) {
  44 bb:
  45   %arrayidx1928 = getelementptr inbounds double** %call1559, i64 %indvars.iv4198
  46   %tmp1888 = load double** %arrayidx1928, align 8
  47   %predphi.v.v = select <4 x i1> %tmp1895, <4 x double> <double -5.000000e-01, double -5.000000e-01, double -5.000000e-01, double -5.000000e-01>, <4 x double> <double 5.000000e-01, double 5.000000e-01, double 5.000000e-01, double 5.000000e-01>
  48   %tmp1900 = bitcast double* %tmp1888 to <4 x double>*
  49   store <4 x double> %predphi.v.v, <4 x double>* %tmp1900, align 8
  50   ret void
  51 }
  52
  53 ; For this test, we used to optimized the conditional mask for the blend, i.e.,
  54 ; we shrunk some of its bits.
  55 ; However, this same mask was used in another select (%predphi31) that turned out
  56 ; to be optimized into a and. In that case, the conditional mask was wrong.
  57 ;
  58 ; Make sure that the and is fed by the original mask.
  59 ;
  60 ; <rdar://problem/18819506>
  61
  62 ; Note: For now, hard code ORIG_MASK and SHRUNK_MASK registers, because we
  63 ; cannot express that ORIG_MASK must not be equal to ORIG_MASK. Otherwise,
  64 ; even a faulty pattern would pass!
  65 ;
  66 ; CHECK-LABEL: test3:
  67 ; Compute the original mask.
  68 ;       CHECK: vpcmpeqd {{%xmm[0-9]+}}, {{%xmm[0-9]+}}, [[ORIG_MASK:%xmm0]]
  69 ; Shrink the bit of the mask.
  70 ; CHECK-NEXT: vpslld    $31, [[ORIG_MASK]], [[SHRUNK_MASK:%xmm3]]
  71 ; Use the shrunk mask in the blend.
  72 ; CHECK-NEXT:   vblendvps       [[SHRUNK_MASK]], %xmm{{[0-9]+}}, %xmm{{[0-9]+}}, %xmm{{[0-9]+}}
  73 ; Use the original mask in the and.
  74 ; CHECK-NEXT: vpand LCPI2_2(%rip), [[ORIG_MASK]], {{%xmm[0-9]+}}
  75 ; CHECK: retq
  76 define void @test3(<4 x i32> %induction30, <4 x i16>* %tmp16, <4 x i16>* %tmp17,  <4 x i16> %tmp3, <4 x i16> %tmp12) {
  77   %tmp6 = srem <4 x i32> %induction30, <i32 3, i32 3, i32 3, i32 3>
  78   %tmp7 = icmp eq <4 x i32> %tmp6, zeroinitializer
  79   %predphi = select <4 x i1> %tmp7, <4 x i16> %tmp3, <4 x i16> %tmp12
  80   %predphi31 = select <4 x i1> %tmp7, <4 x i16> <i16 -1, i16 -1, i16 -1, i16 -1>, <4 x i16> zeroinitializer
  81
  82   store <4 x i16> %predphi31, <4 x i16>* %tmp16, align 8
  83   store <4 x i16> %predphi, <4 x i16>* %tmp17, align 8
  84  ret void
  85 }