llvm/test/Transforms/SLPVectorizer/X86/shuffle-multiple-nodes.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 2
   2 ; RUN: opt -S --passes=slp-vectorizer -mtriple=x86_64-unknown-linux-gnu -mattr=+avx512f < %s | FileCheck %s
   3
   4 define i32 @main(<16 x i32> %bc47.i, <16 x i32> %bc) {
   5 ; CHECK-LABEL: define i32 @main
   6 ; CHECK-SAME: (<16 x i32> [[BC47_I:%.*]], <16 x i32> [[BC:%.*]]) #[[ATTR0:[0-9]+]] {
   7 ; CHECK-NEXT:  entry:
   8 ; CHECK-NEXT:    [[TMP0:%.*]] = shufflevector <16 x i32> [[BC]], <16 x i32> [[BC47_I]], <8 x i32> <i32 0, i32 poison, i32 poison, i32 poison, i32 poison, i32 16, i32 poison, i32 poison>
   9 ; CHECK-NEXT:    [[TMP1:%.*]] = shufflevector <8 x i32> [[TMP0]], <8 x i32> <i32 poison, i32 0, i32 0, i32 0, i32 0, i32 poison, i32 0, i32 0>, <8 x i32> <i32 0, i32 9, i32 10, i32 11, i32 12, i32 5, i32 14, i32 15>
  10 ; CHECK-NEXT:    [[TMP2:%.*]] = shufflevector <8 x i32> [[TMP0]], <8 x i32> <i32 0, i32 0, i32 0, i32 0, i32 0, i32 -1, i32 0, i32 0>, <8 x i32> <i32 poison, i32 0, i32 poison, i32 5, i32 13, i32 poison, i32 poison, i32 poison>
  11 ; CHECK-NEXT:    [[TMP3:%.*]] = shufflevector <8 x i32> <i32 1, i32 undef, i32 0, i32 undef, i32 undef, i32 0, i32 0, i32 0>, <8 x i32> [[TMP2]], <8 x i32> <i32 0, i32 9, i32 2, i32 11, i32 12, i32 5, i32 6, i32 7>
  12 ; CHECK-NEXT:    [[TMP4:%.*]] = mul <8 x i32> [[TMP1]], [[TMP3]]
  13 ; CHECK-NEXT:    [[TMP5:%.*]] = sub <8 x i32> [[TMP4]], zeroinitializer
  14 ; CHECK-NEXT:    [[TMP6:%.*]] = add <8 x i32> [[TMP4]], zeroinitializer
  15 ; CHECK-NEXT:    [[TMP7:%.*]] = shufflevector <8 x i32> [[TMP5]], <8 x i32> [[TMP6]], <8 x i32> <i32 0, i32 9, i32 2, i32 11, i32 4, i32 13, i32 6, i32 15>
  16 ; CHECK-NEXT:    store <8 x i32> [[TMP7]], ptr null, align 16
  17 ; CHECK-NEXT:    ret i32 0
  18 ;
  19 entry:
  20   %0 = extractelement <16 x i32> %bc47.i, i64 0
  21   %1 = extractelement <16 x i32> %bc, i64 0
  22   %2 = extractelement <16 x i32> <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>, i64 0
  23   %3 = mul i32 %1, %2
  24   %4 = mul i32 0, 0
  25   %5 = sub i32 %3, %4
  26   %6 = mul i32 0, 0
  27   %7 = mul i32 0, %1
  28   %8 = add i32 %6, %7
  29   store i32 %5, ptr null, align 16
  30   store i32 %8, ptr getelementptr inbounds ([9 x i32], ptr null, i64 0, i64 1), align 2
  31   %9 = extractelement <16 x i32> zeroinitializer, i64 0
  32   %10 = mul i32 %9, 0
  33   %11 = extractelement <16 x i32> zeroinitializer, i64 0
  34   %12 = mul i32 0, 0
  35   %13 = sub i32 %10, %12
  36   %14 = mul i32 0, %0
  37   %15 = mul i32 0, 0
  38   %16 = add i32 %14, %15
  39   store i32 %13, ptr getelementptr inbounds ([9 x i32], ptr null, i64 0, i64 2), align 4
  40   store i32 %16, ptr getelementptr inbounds ([9 x i32], ptr null, i64 0, i64 3), align 2
  41   %17 = extractelement <16 x i32> <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>, i64 0
  42   %18 = mul i32 %17, 0
  43   %19 = mul i32 0, 0
  44   %20 = sub i32 %18, %19
  45   %21 = extractelement <16 x i32> zeroinitializer, i64 1
  46   %22 = mul i32 %21, %0
  47   %23 = mul i32 0, %17
  48   %24 = add i32 %22, %23
  49   store i32 %20, ptr getelementptr inbounds ([9 x i32], ptr null, i64 0, i64 4), align 8
  50   store i32 %24, ptr getelementptr inbounds ([9 x i32], ptr null, i64 0, i64 5), align 2
  51   %25 = mul i32 0, %11
  52   %26 = extractelement <16 x i32> zeroinitializer, i64 0
  53   %27 = mul i32 0, 0
  54   %28 = sub i32 %25, %27
  55   %29 = mul i32 0, 0
  56   %30 = mul i32 %26, 0
  57   %31 = add i32 %29, %30
  58   store i32 %28, ptr getelementptr inbounds ([9 x i32], ptr null, i64 0, i64 6), align 4
  59   store i32 %31, ptr getelementptr inbounds ([9 x i32], ptr null, i64 0, i64 7), align 2
  60   ret i32 0
  61 }
  62
  63