[SLP][NFC]Add a test with an extra cost of the reused extractelement
authorAlexey Bataev <a.bataev@outlook.com>
Thu, 20 Apr 2023 20:27:18 +0000 (13:27 -0700)
committerAlexey Bataev <a.bataev@outlook.com>
Thu, 20 Apr 2023 20:27:48 +0000 (13:27 -0700)
instruction, NFC.

llvm/test/Transforms/SLPVectorizer/X86/reused-extractelements.ll [new file with mode: 0644]

diff --git a/llvm/test/Transforms/SLPVectorizer/X86/reused-extractelements.ll b/llvm/test/Transforms/SLPVectorizer/X86/reused-extractelements.ll
new file mode 100644 (file)
index 0000000..216db4b
--- /dev/null
@@ -0,0 +1,30 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+; RUN: opt < %s -passes=slp-vectorizer -S -o - -mtriple=x86_64-unknown-linux -mcpu=bdver2 -pass-remarks-output=%t | FileCheck %s
+; RUN: FileCheck --input-file=%t --check-prefix=YAML %s
+
+; YAML: --- !Passed
+; YAML-NEXT: Pass:            slp-vectorizer
+; YAML-NEXT: Name:            VectorizedList
+; YAML-NEXT: Function:        g
+; YAML-NEXT: Args:
+; YAML-NEXT:   - String:          'SLP vectorized with cost '
+; YAML-NEXT:   - Cost:            '-2'
+; YAML-NEXT:   - String:          ' and with tree size '
+; YAML-NEXT:   - TreeSize:        '4'
+
+define <2 x i32> @g(<2 x i32> %x, i32 %a, i32 %b) {
+; CHECK-LABEL: @g(
+; CHECK-NEXT:    [[TMP1:%.*]] = shufflevector <2 x i32> [[X:%.*]], <2 x i32> poison, <2 x i32> <i32 1, i32 undef>
+; CHECK-NEXT:    [[TMP2:%.*]] = insertelement <2 x i32> [[TMP1]], i32 [[A:%.*]], i32 1
+; CHECK-NEXT:    [[TMP3:%.*]] = insertelement <2 x i32> [[TMP1]], i32 [[B:%.*]], i32 1
+; CHECK-NEXT:    [[TMP4:%.*]] = mul <2 x i32> [[TMP2]], [[TMP3]]
+; CHECK-NEXT:    ret <2 x i32> [[TMP4]]
+;
+  %x1 = extractelement <2 x i32> %x, i32 1
+  %x1x1 = mul i32 %x1, %x1
+  %ab = mul i32 %a, %b
+  %ins1 = insertelement <2 x i32> poison, i32 %x1x1, i32 0
+  %ins2 = insertelement <2 x i32> %ins1, i32 %ab, i32 1
+  ret <2 x i32> %ins2
+}
+