test/CodeGen/X86/avx2-conversions.ll

   1 ; RUN: llc < %s -mtriple=x86_64-apple-darwin -mcpu=core-avx2 -mattr=+avx2 | FileCheck %s
   2
   3 define <4 x i32> @trunc4(<4 x i64> %A) nounwind {
   4 ; CHECK-LABEL: trunc4:
   5 ; CHECK:       ## BB#0:
   6 ; CHECK-NEXT:    vmovdqa {{.*#+}} ymm1 = <0,2,4,6,u,u,u,u>
   7 ; CHECK-NEXT:    vpermd %ymm0, %ymm1, %ymm0
   8 ; CHECK-NEXT:    vzeroupper
   9 ; CHECK-NEXT:    retq
  10   %B = trunc <4 x i64> %A to <4 x i32>
  11   ret <4 x i32>%B
  12 }
  13
  14 define <8 x i16> @trunc8(<8 x i32> %A) nounwind {
  15 ; CHECK-LABEL: trunc8:
  16 ; CHECK:       ## BB#0:
  17 ; CHECK-NEXT:    vpshufb {{.*#+}} ymm0 = ymm0[0,1,4,5,8,9,12,13],zero,zero,zero,zero,zero,zero,zero,zero,ymm0[16,17,20,21,24,25,28,29],zero,zero,zero,zero,zero,zero,zero,zero
  18 ; CHECK-NEXT:    vpermq {{.*#+}} ymm0 = ymm0[0,2,2,3]
  19 ; CHECK-NEXT:    vzeroupper
  20 ; CHECK-NEXT:    retq
  21   %B = trunc <8 x i32> %A to <8 x i16>
  22   ret <8 x i16>%B
  23 }
  24
  25 define <4 x i64> @sext4(<4 x i32> %A) nounwind {
  26 ; CHECK-LABEL: sext4:
  27 ; CHECK:       ## BB#0:
  28 ; CHECK-NEXT:    vpmovsxdq %xmm0, %ymm0
  29 ; CHECK-NEXT:    retq
  30   %B = sext <4 x i32> %A to <4 x i64>
  31   ret <4 x i64>%B
  32 }
  33
  34 define <8 x i32> @sext8(<8 x i16> %A) nounwind {
  35 ; CHECK-LABEL: sext8:
  36 ; CHECK:       ## BB#0:
  37 ; CHECK-NEXT:    vpmovsxwd %xmm0, %ymm0
  38 ; CHECK-NEXT:    retq
  39   %B = sext <8 x i16> %A to <8 x i32>
  40   ret <8 x i32>%B
  41 }
  42
  43 define <4 x i64> @zext4(<4 x i32> %A) nounwind {
  44 ; CHECK-LABEL: zext4:
  45 ; CHECK:       ## BB#0:
  46 ; CHECK-NEXT:    vpmovzxdq {{.*#+}} ymm0 = xmm0[0],zero,xmm0[1],zero,xmm0[2],zero,xmm0[3],zero
  47 ; CHECK-NEXT:    retq
  48   %B = zext <4 x i32> %A to <4 x i64>
  49   ret <4 x i64>%B
  50 }
  51
  52 define <8 x i32> @zext8(<8 x i16> %A) nounwind {
  53 ; CHECK-LABEL: zext8:
  54 ; CHECK:       ## BB#0:
  55 ; CHECK-NEXT:    vpmovzxwd {{.*#+}} ymm0 = xmm0[0],zero,xmm0[1],zero,xmm0[2],zero,xmm0[3],zero,xmm0[4],zero,xmm0[5],zero,xmm0[6],zero,xmm0[7],zero
  56 ; CHECK-NEXT:    retq
  57   %B = zext <8 x i16> %A to <8 x i32>
  58   ret <8 x i32>%B
  59 }
  60
  61 define <8 x i32> @zext_8i8_8i32(<8 x i8> %A) nounwind {
  62 ; CHECK-LABEL: zext_8i8_8i32:
  63 ; CHECK:       ## BB#0:
  64 ; CHECK-NEXT:    vpand {{.*}}(%rip), %xmm0, %xmm0
  65 ; CHECK-NEXT:    vpmovzxwd {{.*#+}} ymm0 = xmm0[0],zero,xmm0[1],zero,xmm0[2],zero,xmm0[3],zero,xmm0[4],zero,xmm0[5],zero,xmm0[6],zero,xmm0[7],zero
  66 ; CHECK-NEXT:    retq
  67   %B = zext <8 x i8> %A to <8 x i32>
  68   ret <8 x i32>%B
  69 }
  70
  71 define <16 x i16> @zext_16i8_16i16(<16 x i8> %z) {
  72 ; CHECK-LABEL: zext_16i8_16i16:
  73 ; CHECK:       ## BB#0:
  74 ; CHECK-NEXT:    vpmovzxbw {{.*#+}} ymm0 = xmm0[0],zero,xmm0[1],zero,xmm0[2],zero,xmm0[3],zero,xmm0[4],zero,xmm0[5],zero,xmm0[6],zero,xmm0[7],zero,xmm0[8],zero,xmm0[9],zero,xmm0[10],zero,xmm0[11],zero,xmm0[12],zero,xmm0[13],zero,xmm0[14],zero,xmm0[15],zero
  75 ; CHECK-NEXT:    retq
  76   %t = zext <16 x i8> %z to <16 x i16>
  77   ret <16 x i16> %t
  78 }
  79
  80 define <16 x i16> @sext_16i8_16i16(<16 x i8> %z) {
  81 ; CHECK-LABEL: sext_16i8_16i16:
  82 ; CHECK:       ## BB#0:
  83 ; CHECK-NEXT:    vpmovsxbw %xmm0, %ymm0
  84 ; CHECK-NEXT:    retq
  85   %t = sext <16 x i8> %z to <16 x i16>
  86   ret <16 x i16> %t
  87 }
  88
  89 define <16 x i8> @trunc_16i16_16i8(<16 x i16> %z) {
  90 ; CHECK-LABEL: trunc_16i16_16i8:
  91 ; CHECK:       ## BB#0:
  92 ; CHECK-NEXT:    vextracti128 $1, %ymm0, %xmm1
  93 ; CHECK-NEXT:    vmovdqa {{.*#+}} xmm2 = <0,2,4,6,8,10,12,14,u,u,u,u,u,u,u,u>
  94 ; CHECK-NEXT:    vpshufb %xmm2, %xmm1, %xmm1
  95 ; CHECK-NEXT:    vpshufb %xmm2, %xmm0, %xmm0
  96 ; CHECK-NEXT:    vpunpcklqdq {{.*#+}} xmm0 = xmm0[0],xmm1[0]
  97 ; CHECK-NEXT:    vzeroupper
  98 ; CHECK-NEXT:    retq
  99   %t = trunc <16 x i16> %z to <16 x i8>
 100   ret <16 x i8> %t
 101 }
 102
 103 define <4 x i64> @load_sext_test1(<4 x i32> *%ptr) {
 104 ; CHECK-LABEL: load_sext_test1:
 105 ; CHECK:       ## BB#0:
 106 ; CHECK-NEXT:    vpmovsxdq (%rdi), %ymm0
 107 ; CHECK-NEXT:    retq
 108  %X = load <4 x i32>, <4 x i32>* %ptr
 109  %Y = sext <4 x i32> %X to <4 x i64>
 110  ret <4 x i64>%Y
 111 }
 112
 113 define <4 x i64> @load_sext_test2(<4 x i8> *%ptr) {
 114 ; CHECK-LABEL: load_sext_test2:
 115 ; CHECK:       ## BB#0:
 116 ; CHECK-NEXT:    vpmovsxbq (%rdi), %ymm0
 117 ; CHECK-NEXT:    retq
 118  %X = load <4 x i8>, <4 x i8>* %ptr
 119  %Y = sext <4 x i8> %X to <4 x i64>
 120  ret <4 x i64>%Y
 121 }
 122
 123 define <4 x i64> @load_sext_test3(<4 x i16> *%ptr) {
 124 ; CHECK-LABEL: load_sext_test3:
 125 ; CHECK:       ## BB#0:
 126 ; CHECK-NEXT:    vpmovsxwq (%rdi), %ymm0
 127 ; CHECK-NEXT:    retq
 128  %X = load <4 x i16>, <4 x i16>* %ptr
 129  %Y = sext <4 x i16> %X to <4 x i64>
 130  ret <4 x i64>%Y
 131 }
 132
 133 define <8 x i32> @load_sext_test4(<8 x i16> *%ptr) {
 134 ; CHECK-LABEL: load_sext_test4:
 135 ; CHECK:       ## BB#0:
 136 ; CHECK-NEXT:    vpmovsxwd (%rdi), %ymm0
 137 ; CHECK-NEXT:    retq
 138  %X = load <8 x i16>, <8 x i16>* %ptr
 139  %Y = sext <8 x i16> %X to <8 x i32>
 140  ret <8 x i32>%Y
 141 }
 142
 143 define <8 x i32> @load_sext_test5(<8 x i8> *%ptr) {
 144 ; CHECK-LABEL: load_sext_test5:
 145 ; CHECK:       ## BB#0:
 146 ; CHECK-NEXT:    vpmovsxbd (%rdi), %ymm0
 147 ; CHECK-NEXT:    retq
 148  %X = load <8 x i8>, <8 x i8>* %ptr
 149  %Y = sext <8 x i8> %X to <8 x i32>
 150  ret <8 x i32>%Y
 151 }