test/CodeGen/X86/avx512-shift.ll

   1 ;RUN: llc < %s -mtriple=x86_64-apple-darwin -mcpu=knl | FileCheck %s
   2 ;RUN: llc < %s -mtriple=x86_64-apple-darwin -mcpu=skx | FileCheck --check-prefix=SKX %s
   3
   4 ;CHECK-LABEL: shift_16_i32
   5 ;CHECK: vpsrld
   6 ;CHECK: vpslld
   7 ;CHECK: vpsrad
   8 ;CHECK: ret
   9 define <16 x i32> @shift_16_i32(<16 x i32> %a) {
  10    %b = lshr <16 x i32> %a, <i32 1, i32 1, i32 1, i32 1, i32 1, i32 1, i32 1, i32 1, i32 1, i32 1, i32 1, i32 1, i32 1, i32 1, i32 1, i32 1>
  11    %c = shl <16 x i32> %b, <i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12>
  12    %d = ashr <16 x i32> %c, <i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12, i32 12>
  13    ret <16 x i32> %d;
  14 }
  15
  16 ;CHECK-LABEL: shift_8_i64
  17 ;CHECK: vpsrlq
  18 ;CHECK: vpsllq
  19 ;CHECK: vpsraq
  20 ;CHECK: ret
  21 define <8 x i64> @shift_8_i64(<8 x i64> %a) {
  22    %b = lshr <8 x i64> %a, <i64 1, i64 1, i64 1, i64 1, i64 1, i64 1, i64 1, i64 1>
  23    %c = shl <8 x i64> %b,  <i64 12, i64 12, i64 12, i64 12, i64 12, i64 12, i64 12, i64 12>
  24    %d = ashr <8 x i64> %c, <i64 12, i64 12, i64 12, i64 12, i64 12, i64 12, i64 12, i64 12>
  25    ret <8 x i64> %d;
  26 }
  27
  28 ;SKX-LABEL: shift_4_i64
  29 ;SKX: vpsrlq
  30 ;SKX: vpsllq
  31 ;SKX: vpsraq
  32 ;SKX: ret
  33 define <4 x i64> @shift_4_i64(<4 x i64> %a) {
  34    %b = lshr <4 x i64> %a, <i64 1, i64 1, i64 1, i64 1>
  35    %c = shl <4 x i64> %b,  <i64 12, i64 12, i64 12, i64 12>
  36    %d = ashr <4 x i64> %c, <i64 12, i64 12, i64 12, i64 12>
  37    ret <4 x i64> %d;
  38 }
  39
  40 ; CHECK-LABEL: variable_shl4
  41 ; CHECK: vpsllvq %zmm
  42 ; CHECK: ret
  43 define <8 x i64> @variable_shl4(<8 x i64> %x, <8 x i64> %y) {
  44   %k = shl <8 x i64> %x, %y
  45   ret <8 x i64> %k
  46 }
  47
  48 ; CHECK-LABEL: variable_shl5
  49 ; CHECK: vpsllvd %zmm
  50 ; CHECK: ret
  51 define <16 x i32> @variable_shl5(<16 x i32> %x, <16 x i32> %y) {
  52   %k = shl <16 x i32> %x, %y
  53   ret <16 x i32> %k
  54 }
  55
  56 ; CHECK-LABEL: variable_srl0
  57 ; CHECK: vpsrlvd
  58 ; CHECK: ret
  59 define <16 x i32> @variable_srl0(<16 x i32> %x, <16 x i32> %y) {
  60   %k = lshr <16 x i32> %x, %y
  61   ret <16 x i32> %k
  62 }
  63
  64 ; CHECK-LABEL: variable_srl2
  65 ; CHECK: psrlvq
  66 ; CHECK: ret
  67 define <8 x i64> @variable_srl2(<8 x i64> %x, <8 x i64> %y) {
  68   %k = lshr <8 x i64> %x, %y
  69   ret <8 x i64> %k
  70 }
  71
  72 ; CHECK-LABEL: variable_sra1
  73 ; CHECK: vpsravd
  74 ; CHECK: ret
  75 define <16 x i32> @variable_sra1(<16 x i32> %x, <16 x i32> %y) {
  76   %k = ashr <16 x i32> %x, %y
  77   ret <16 x i32> %k
  78 }
  79
  80 ; CHECK-LABEL: variable_sra2
  81 ; CHECK: vpsravq %zmm
  82 ; CHECK: ret
  83 define <8 x i64> @variable_sra2(<8 x i64> %x, <8 x i64> %y) {
  84   %k = ashr <8 x i64> %x, %y
  85   ret <8 x i64> %k
  86 }
  87
  88 ; SKX-LABEL: variable_sra3
  89 ; SKX: vpsravq %ymm
  90 ; SKX: ret
  91 define <4 x i64> @variable_sra3(<4 x i64> %x, <4 x i64> %y) {
  92   %k = ashr <4 x i64> %x, %y
  93   ret <4 x i64> %k
  94 }
  95
  96 ; SKX-LABEL: variable_sra4
  97 ; SKX: vpsravw %xmm
  98 ; SKX: ret
  99 define <8 x i16> @variable_sra4(<8 x i16> %x, <8 x i16> %y) {
 100   %k = ashr <8 x i16> %x, %y
 101   ret <8 x i16> %k
 102 }
 103
 104 ; CHECK-LABEL: variable_sra01_load
 105 ; CHECK: vpsravd (%
 106 ; CHECK: ret
 107 define <16 x i32> @variable_sra01_load(<16 x i32> %x, <16 x i32>* %y) {
 108   %y1 = load <16 x i32>, <16 x i32>* %y
 109   %k = ashr <16 x i32> %x, %y1
 110   ret <16 x i32> %k
 111 }
 112
 113 ; CHECK-LABEL: variable_shl1_load
 114 ; CHECK: vpsllvd (%
 115 ; CHECK: ret
 116 define <16 x i32> @variable_shl1_load(<16 x i32> %x, <16 x i32>* %y) {
 117   %y1 = load <16 x i32>, <16 x i32>* %y
 118   %k = shl <16 x i32> %x, %y1
 119   ret <16 x i32> %k
 120 }
 121 ; CHECK: variable_srl0_load
 122 ; CHECK: vpsrlvd (%
 123 ; CHECK: ret
 124 define <16 x i32> @variable_srl0_load(<16 x i32> %x, <16 x i32>* %y) {
 125   %y1 = load <16 x i32>, <16 x i32>* %y
 126   %k = lshr <16 x i32> %x, %y1
 127   ret <16 x i32> %k
 128 }
 129
 130 ; CHECK: variable_srl3_load
 131 ; CHECK: vpsrlvq (%
 132 ; CHECK: ret
 133 define <8 x i64> @variable_srl3_load(<8 x i64> %x, <8 x i64>* %y) {
 134   %y1 = load <8 x i64>, <8 x i64>* %y
 135   %k = lshr <8 x i64> %x, %y1
 136   ret <8 x i64> %k
 137 }