[SLSR] handle candidate form (B + i * S)
[oota-llvm.git] / test / Transforms / StraightLineStrengthReduce / X86 / no-slsr.ll
1 ; RUN: opt < %s -slsr -gvn -dce -S | FileCheck %s
2
3 target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
4 target triple = "x86_64-unknown-linux-gnu"
5
6 ; Do not perform SLSR on &input[s] and &input[s * 2] which fit into addressing
7 ; modes of X86.
8 define i32 @no_slsr_gep(i32* %input, i64 %s) {
9 ; CHECK-LABEL: @no_slsr_gep(
10   ; v0 = input[0];
11   %p0 = getelementptr inbounds i32, i32* %input, i64 0
12   %v0 = load i32, i32* %p0
13
14   ; v1 = input[s];
15   %p1 = getelementptr inbounds i32, i32* %input, i64 %s
16 ; CHECK: %p1 = getelementptr inbounds i32, i32* %input, i64 %s
17   %v1 = load i32, i32* %p1
18
19   ; v2 = input[s * 2];
20   %s2 = mul nsw i64 %s, 2
21   %p2 = getelementptr inbounds i32, i32* %input, i64 %s2
22 ; CHECK: %p2 = getelementptr inbounds i32, i32* %input, i64 %s2
23   %v2 = load i32, i32* %p2
24
25   ; return v0 + v1 + v2;
26   %1 = add i32 %v0, %v1
27   %2 = add i32 %1, %v2
28   ret i32 %2
29 }
30
31 define void @no_slsr_add(i32 %b, i32 %s) {
32 ; CHECK-LABEL: @no_slsr_add(
33   %1 = add i32 %b, %s
34 ; CHECK: add i32 %b, %s
35   call void @foo(i32 %1)
36   %s2 = mul i32 %s, 2
37 ; CHECK: %s2 = mul i32 %s, 2
38   %2 = add i32 %b, %s2
39 ; CHECK: add i32 %b, %s2
40   call void @foo(i32 %2)
41   ret void
42 }
43
44 declare void @foo(i32 %a)