test/CodeGen/X86/reduce-trunc-shl.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc < %s -mtriple=x86_64-unknown-linux-gnu -mattr=+sse2 | FileCheck %s --check-prefix=SSE2
   3 ; RUN: llc < %s -mtriple=x86_64-unknown-linux-gnu -mattr=+avx2 | FileCheck %s --check-prefix=AVX2
   4
   5 define void @trunc_shl_7_v4i32_v4i64(<4 x i32> addrspace(1)* %out, <4 x i64> addrspace(1)* %in) {
   6 ; SSE2-LABEL: trunc_shl_7_v4i32_v4i64:
   7 ; SSE2:       # %bb.0:
   8 ; SSE2-NEXT:    movaps (%rsi), %xmm0
   9 ; SSE2-NEXT:    shufps {{.*#+}} xmm0 = xmm0[0,2],mem[0,2]
  10 ; SSE2-NEXT:    pslld $7, %xmm0
  11 ; SSE2-NEXT:    movdqa %xmm0, (%rdi)
  12 ; SSE2-NEXT:    retq
  13 ;
  14 ; AVX2-LABEL: trunc_shl_7_v4i32_v4i64:
  15 ; AVX2:       # %bb.0:
  16 ; AVX2-NEXT:    vpshufd {{.*#+}} ymm0 = mem[0,2,2,3,4,6,6,7]
  17 ; AVX2-NEXT:    vpermq {{.*#+}} ymm0 = ymm0[0,2,2,3]
  18 ; AVX2-NEXT:    vpslld $7, %xmm0, %xmm0
  19 ; AVX2-NEXT:    vmovdqa %xmm0, (%rdi)
  20 ; AVX2-NEXT:    vzeroupper
  21 ; AVX2-NEXT:    retq
  22   %val = load <4 x i64>, <4 x i64> addrspace(1)* %in
  23   %shl = shl <4 x i64> %val, <i64 7, i64 7, i64 7, i64 7>
  24   %trunc = trunc <4 x i64> %shl to <4 x i32>
  25   store <4 x i32> %trunc, <4 x i32> addrspace(1)* %out
  26   ret void
  27 }
  28
  29 define <8 x i16> @trunc_shl_15_v8i16_v8i32(<8 x i32> %a) {
  30 ; SSE2-LABEL: trunc_shl_15_v8i16_v8i32:
  31 ; SSE2:       # %bb.0:
  32 ; SSE2-NEXT:    pslld $16, %xmm1
  33 ; SSE2-NEXT:    psrad $16, %xmm1
  34 ; SSE2-NEXT:    pslld $16, %xmm0
  35 ; SSE2-NEXT:    psrad $16, %xmm0
  36 ; SSE2-NEXT:    packssdw %xmm1, %xmm0
  37 ; SSE2-NEXT:    psllw $15, %xmm0
  38 ; SSE2-NEXT:    retq
  39 ;
  40 ; AVX2-LABEL: trunc_shl_15_v8i16_v8i32:
  41 ; AVX2:       # %bb.0:
  42 ; AVX2-NEXT:    vpshufb {{.*#+}} ymm0 = ymm0[0,1,4,5,8,9,12,13,8,9,12,13,12,13,14,15,16,17,20,21,24,25,28,29,24,25,28,29,28,29,30,31]
  43 ; AVX2-NEXT:    vpermq {{.*#+}} ymm0 = ymm0[0,2,2,3]
  44 ; AVX2-NEXT:    vpsllw $15, %xmm0, %xmm0
  45 ; AVX2-NEXT:    vzeroupper
  46 ; AVX2-NEXT:    retq
  47   %shl = shl <8 x i32> %a, <i32 15, i32 15, i32 15, i32 15, i32 15, i32 15, i32 15, i32 15>
  48   %conv = trunc <8 x i32> %shl to <8 x i16>
  49   ret <8 x i16> %conv
  50 }
  51
  52 define <8 x i16> @trunc_shl_16_v8i16_v8i32(<8 x i32> %a) {
  53 ; SSE2-LABEL: trunc_shl_16_v8i16_v8i32:
  54 ; SSE2:       # %bb.0:
  55 ; SSE2-NEXT:    xorps %xmm0, %xmm0
  56 ; SSE2-NEXT:    retq
  57 ;
  58 ; AVX2-LABEL: trunc_shl_16_v8i16_v8i32:
  59 ; AVX2:       # %bb.0:
  60 ; AVX2-NEXT:    vxorps %xmm0, %xmm0, %xmm0
  61 ; AVX2-NEXT:    retq
  62   %shl = shl <8 x i32> %a, <i32 16, i32 16, i32 16, i32 16, i32 16, i32 16, i32 16, i32 16>
  63   %conv = trunc <8 x i32> %shl to <8 x i16>
  64   ret <8 x i16> %conv
  65 }
  66
  67 define <8 x i16> @trunc_shl_17_v8i16_v8i32(<8 x i32> %a) {
  68 ; SSE2-LABEL: trunc_shl_17_v8i16_v8i32:
  69 ; SSE2:       # %bb.0:
  70 ; SSE2-NEXT:    xorps %xmm0, %xmm0
  71 ; SSE2-NEXT:    retq
  72 ;
  73 ; AVX2-LABEL: trunc_shl_17_v8i16_v8i32:
  74 ; AVX2:       # %bb.0:
  75 ; AVX2-NEXT:    vxorps %xmm0, %xmm0, %xmm0
  76 ; AVX2-NEXT:    retq
  77   %shl = shl <8 x i32> %a, <i32 17, i32 17, i32 17, i32 17, i32 17, i32 17, i32 17, i32 17>
  78   %conv = trunc <8 x i32> %shl to <8 x i16>
  79   ret <8 x i16> %conv
  80 }
  81
  82 define void @trunc_shl_31_i32_i64(i32* %out, i64* %in) {
  83 ; SSE2-LABEL: trunc_shl_31_i32_i64:
  84 ; SSE2:       # %bb.0:
  85 ; SSE2-NEXT:    movl (%rsi), %eax
  86 ; SSE2-NEXT:    shll $31, %eax
  87 ; SSE2-NEXT:    movl %eax, (%rdi)
  88 ; SSE2-NEXT:    retq
  89 ;
  90 ; AVX2-LABEL: trunc_shl_31_i32_i64:
  91 ; AVX2:       # %bb.0:
  92 ; AVX2-NEXT:    movl (%rsi), %eax
  93 ; AVX2-NEXT:    shll $31, %eax
  94 ; AVX2-NEXT:    movl %eax, (%rdi)
  95 ; AVX2-NEXT:    retq
  96   %val = load i64, i64* %in
  97   %shl = shl i64 %val, 31
  98   %trunc = trunc i64 %shl to i32
  99   store i32 %trunc, i32* %out
 100   ret void
 101 }
 102
 103 define void @trunc_shl_32_i32_i64(i32* %out, i64* %in) {
 104 ; SSE2-LABEL: trunc_shl_32_i32_i64:
 105 ; SSE2:       # %bb.0:
 106 ; SSE2-NEXT:    movl $0, (%rdi)
 107 ; SSE2-NEXT:    retq
 108 ;
 109 ; AVX2-LABEL: trunc_shl_32_i32_i64:
 110 ; AVX2:       # %bb.0:
 111 ; AVX2-NEXT:    movl $0, (%rdi)
 112 ; AVX2-NEXT:    retq
 113   %val = load i64, i64* %in
 114   %shl = shl i64 %val, 32
 115   %trunc = trunc i64 %shl to i32
 116   store i32 %trunc, i32* %out
 117   ret void
 118 }
 119
 120 define void @trunc_shl_15_i16_i64(i16* %out, i64* %in) {
 121 ; SSE2-LABEL: trunc_shl_15_i16_i64:
 122 ; SSE2:       # %bb.0:
 123 ; SSE2-NEXT:    movl (%rsi), %eax
 124 ; SSE2-NEXT:    shll $15, %eax
 125 ; SSE2-NEXT:    movw %ax, (%rdi)
 126 ; SSE2-NEXT:    retq
 127 ;
 128 ; AVX2-LABEL: trunc_shl_15_i16_i64:
 129 ; AVX2:       # %bb.0:
 130 ; AVX2-NEXT:    movl (%rsi), %eax
 131 ; AVX2-NEXT:    shll $15, %eax
 132 ; AVX2-NEXT:    movw %ax, (%rdi)
 133 ; AVX2-NEXT:    retq
 134   %val = load i64, i64* %in
 135   %shl = shl i64 %val, 15
 136   %trunc = trunc i64 %shl to i16
 137   store i16 %trunc, i16* %out
 138   ret void
 139 }
 140
 141 define void @trunc_shl_16_i16_i64(i16* %out, i64* %in) {
 142 ; SSE2-LABEL: trunc_shl_16_i16_i64:
 143 ; SSE2:       # %bb.0:
 144 ; SSE2-NEXT:    movw $0, (%rdi)
 145 ; SSE2-NEXT:    retq
 146 ;
 147 ; AVX2-LABEL: trunc_shl_16_i16_i64:
 148 ; AVX2:       # %bb.0:
 149 ; AVX2-NEXT:    movw $0, (%rdi)
 150 ; AVX2-NEXT:    retq
 151   %val = load i64, i64* %in
 152   %shl = shl i64 %val, 16
 153   %trunc = trunc i64 %shl to i16
 154   store i16 %trunc, i16* %out
 155   ret void
 156 }
 157
 158 define void @trunc_shl_7_i8_i64(i8* %out, i64* %in) {
 159 ; SSE2-LABEL: trunc_shl_7_i8_i64:
 160 ; SSE2:       # %bb.0:
 161 ; SSE2-NEXT:    movb (%rsi), %al
 162 ; SSE2-NEXT:    shlb $7, %al
 163 ; SSE2-NEXT:    movb %al, (%rdi)
 164 ; SSE2-NEXT:    retq
 165 ;
 166 ; AVX2-LABEL: trunc_shl_7_i8_i64:
 167 ; AVX2:       # %bb.0:
 168 ; AVX2-NEXT:    movb (%rsi), %al
 169 ; AVX2-NEXT:    shlb $7, %al
 170 ; AVX2-NEXT:    movb %al, (%rdi)
 171 ; AVX2-NEXT:    retq
 172   %val = load i64, i64* %in
 173   %shl = shl i64 %val, 7
 174   %trunc = trunc i64 %shl to i8
 175   store i8 %trunc, i8* %out
 176   ret void
 177 }
 178
 179 define void @trunc_shl_8_i8_i64(i8* %out, i64* %in) {
 180 ; SSE2-LABEL: trunc_shl_8_i8_i64:
 181 ; SSE2:       # %bb.0:
 182 ; SSE2-NEXT:    movb $0, (%rdi)
 183 ; SSE2-NEXT:    retq
 184 ;
 185 ; AVX2-LABEL: trunc_shl_8_i8_i64:
 186 ; AVX2:       # %bb.0:
 187 ; AVX2-NEXT:    movb $0, (%rdi)
 188 ; AVX2-NEXT:    retq
 189   %val = load i64, i64* %in
 190   %shl = shl i64 %val, 8
 191   %trunc = trunc i64 %shl to i8
 192   store i8 %trunc, i8* %out
 193   ret void
 194 }