test/CodeGen/X86/swizzle-avx2.ll

   1 ; RUN: llc < %s -march=x86-64 -mtriple=x86_64-unknown-linux-gnu -mcpu=core-avx2 | FileCheck %s
   2
   3 ; Test that we correctly fold a shuffle that performs a swizzle of another
   4 ; shuffle node according to the rule
   5 ;  shuffle (shuffle (x, undef, M0), undef, M1) -> shuffle(x, undef, M2)
   6 ;
   7 ; We only do this if the resulting mask is legal to avoid introducing an
   8 ; illegal shuffle that is expanded into a sub-optimal sequence of instructions
   9 ; during lowering stage.
  10
  11 ; Check that we produce a single vector permute / shuffle in all cases.
  12
  13 define <8 x i32> @swizzle_1(<8 x i32> %v) {
  14   %1 = shufflevector <8 x i32> %v, <8 x i32> undef, <8 x i32> <i32 3, i32 1, i32 2, i32 0, i32 7, i32 5, i32 6, i32 4>
  15   %2 = shufflevector <8 x i32> %1, <8 x i32> undef, <8 x i32> <i32 1, i32 0, i32 2, i32 3, i32 7, i32 5, i32 6, i32 4>
  16   ret <8 x i32> %2
  17 }
  18 ; CHECK-LABEL: swizzle_1
  19 ; CHECK: vpermd
  20 ; CHECK-NOT: vpermd
  21 ; CHECK: ret
  22
  23
  24 define <8 x i32> @swizzle_2(<8 x i32> %v) {
  25   %1 = shufflevector <8 x i32> %v, <8 x i32> undef, <8 x i32> <i32 6, i32 7, i32 4, i32 5, i32 0, i32 1, i32 2, i32 3>
  26   %2 = shufflevector <8 x i32> %1, <8 x i32> undef, <8 x i32> <i32 6, i32 7, i32 4, i32 5, i32 0, i32 1, i32 2, i32 3>
  27   ret <8 x i32> %2
  28 }
  29 ; CHECK-LABEL: swizzle_2
  30 ; CHECK: vpshufd $78
  31 ; CHECK-NOT: vpermd
  32 ; CHECK-NOT: vpshufd
  33 ; CHECK: ret
  34
  35
  36 define <8 x i32> @swizzle_3(<8 x i32> %v) {
  37   %1 = shufflevector <8 x i32> %v, <8 x i32> undef, <8 x i32> <i32 4, i32 5, i32 6, i32 7, i32 2, i32 3, i32 0, i32 1>
  38   %2 = shufflevector <8 x i32> %1, <8 x i32> undef, <8 x i32> <i32 4, i32 5, i32 6, i32 7, i32 2, i32 3, i32 0, i32 1>
  39   ret <8 x i32> %2
  40 }
  41 ; CHECK-LABEL: swizzle_3
  42 ; CHECK: vpshufd $78
  43 ; CHECK-NOT: vpermd
  44 ; CHECK-NOT: vpshufd
  45 ; CHECK: ret
  46
  47
  48 define <8 x i32> @swizzle_4(<8 x i32> %v) {
  49   %1 = shufflevector <8 x i32> %v, <8 x i32> undef, <8 x i32> <i32 4, i32 7, i32 5, i32 6, i32 3, i32 2, i32 0, i32 1>
  50   %2 = shufflevector <8 x i32> %1, <8 x i32> undef, <8 x i32> <i32 4, i32 7, i32 5, i32 6, i32 3, i32 2, i32 0, i32 1>
  51   ret <8 x i32> %2
  52 }
  53 ; CHECK-LABEL: swizzle_4
  54 ; CHECK: vpermd
  55 ; CHECK-NOT: vpermd
  56 ; CHECK: ret
  57
  58
  59 define <8 x i32> @swizzle_5(<8 x i32> %v) {
  60   %1 = shufflevector <8 x i32> %v, <8 x i32> undef, <8 x i32> <i32 7, i32 4, i32 6, i32 5, i32 0, i32 2, i32 1, i32 3>
  61   %2 = shufflevector <8 x i32> %1, <8 x i32> undef, <8 x i32> <i32 7, i32 4, i32 6, i32 5, i32 0, i32 2, i32 1, i32 3>
  62   ret <8 x i32> %2
  63 }
  64 ; CHECK-LABEL: swizzle_5
  65 ; CHECK: vpermd
  66 ; CHECK-NOT: vpermd
  67 ; CHECK: ret
  68
  69
  70 define <8 x i32> @swizzle_6(<8 x i32> %v) {
  71   %1 = shufflevector <8 x i32> %v, <8 x i32> undef, <8 x i32> <i32 2, i32 1, i32 3, i32 0, i32 4, i32 7, i32 6, i32 5>
  72   %2 = shufflevector <8 x i32> %1, <8 x i32> undef, <8 x i32> <i32 2, i32 1, i32 3, i32 0, i32 4, i32 7, i32 6, i32 5>
  73   ret <8 x i32> %2
  74 }
  75 ; CHECK-LABEL: swizzle_6
  76 ; CHECK: vpermd
  77 ; CHECK-NOT: vpermd
  78 ; CHECK: ret
  79
  80
  81 define <8 x i32> @swizzle_7(<8 x i32> %v) {
  82   %1 = shufflevector <8 x i32> %v, <8 x i32> undef, <8 x i32> <i32 0, i32 3, i32 1, i32 2, i32 5, i32 4, i32 6, i32 7>
  83   %2 = shufflevector <8 x i32> %1, <8 x i32> undef, <8 x i32> <i32 0, i32 3, i32 1, i32 2, i32 5, i32 4, i32 6, i32 7>
  84   ret <8 x i32> %2
  85 }
  86 ; CHECK-LABEL: swizzle_7
  87 ; CHECK: vpermd
  88 ; CHECK-NOT: vpermd
  89 ; CHECK: ret
  90
  91