[llvm] d3a2436 - [X86] Add test for foldable add of mul (NFC)

[Nikita Popov via llvm-commits]

Author: Nikita Popov
Date: 2023-04-14T15:57:30+02:00
New Revision: d3a2436b82a6b25d7998a2be11270e4601a8f09d

URL: https://github.com/llvm/llvm-project/commit/d3a2436b82a6b25d7998a2be11270e4601a8f09d
DIFF: https://github.com/llvm/llvm-project/commit/d3a2436b82a6b25d7998a2be11270e4601a8f09d.diff

LOG: [X86] Add test for foldable add of mul (NFC)

Added: 
    llvm/test/CodeGen/X86/add-of-mul.ll

Modified: 
    

Removed: 
    


################################################################################
diff  --git a/llvm/test/CodeGen/X86/add-of-mul.ll b/llvm/test/CodeGen/X86/add-of-mul.ll
new file mode 100644
index 000000000000..843d397ee7c4
--- /dev/null
+++ b/llvm/test/CodeGen/X86/add-of-mul.ll
@@ -0,0 +1,63 @@
+; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 2
+; RUN: llc < %s -mtriple=x86_64-unknown-linux-gnu | FileCheck %s
+
+define i32 @test_scalar(i32 %x) {
+; CHECK-LABEL: test_scalar:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    # kill: def $edi killed $edi def $rdi
+; CHECK-NEXT:    leal (%rdi,%rdi,2), %eax
+; CHECK-NEXT:    addl %edi, %eax
+; CHECK-NEXT:    retq
+  %mul = mul i32 %x, 3
+  %add = add i32 %mul, %x
+  ret i32 %add
+}
+
+define <4 x i32> @test_vector(<4 x i32> %x) {
+; CHECK-LABEL: test_vector:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    paddd %xmm0, %xmm0
+; CHECK-NEXT:    paddd %xmm0, %xmm0
+; CHECK-NEXT:    retq
+  %mul = mul <4 x i32> %x, <i32 3, i32 3, i32 3, i32 3>
+  %add = add <4 x i32> %mul, %x
+  ret <4 x i32> %add
+}
+
+define ptr @test_ptr(ptr %p) {
+; CHECK-LABEL: test_ptr:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    leaq (%rdi,%rdi,2), %rax
+; CHECK-NEXT:    addq %rdi, %rax
+; CHECK-NEXT:    retq
+  %addr = ptrtoint ptr %p to i64
+  %mul = mul i64 %addr, 3
+  %res = getelementptr i8, ptr %p, i64 %mul
+  ret ptr %res
+}
+
+define i32 @test_extra_use(i32 %x, ptr %p) {
+; CHECK-LABEL: test_extra_use:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    # kill: def $edi killed $edi def $rdi
+; CHECK-NEXT:    leal (%rdi,%rdi,2), %eax
+; CHECK-NEXT:    movl %eax, (%rsi)
+; CHECK-NEXT:    addl %edi, %eax
+; CHECK-NEXT:    retq
+  %mul = mul i32 %x, 3
+  store i32 %mul, ptr %p
+  %add = add i32 %mul, %x
+  ret i32 %add
+}
+
+define i32 @test_
diff erent_operand(i32 %x, i32 %y) {
+; CHECK-LABEL: test_
diff erent_operand:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    # kill: def $edi killed $edi def $rdi
+; CHECK-NEXT:    leal (%rdi,%rdi,2), %eax
+; CHECK-NEXT:    addl %esi, %eax
+; CHECK-NEXT:    retq
+  %mul = mul i32 %x, 3
+  %add = add i32 %mul, %y
+  ret i32 %add
+}