1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2; RUN: opt < %s -reassociate -S | FileCheck %s
3
4; Check that a*a*b+a*a*c is turned into a*(a*(b+c)).
5
6define i64 @multistep1(i64 %a, i64 %b, i64 %c) {
7; CHECK-LABEL: @multistep1(
8; CHECK-NEXT:    [[REASS_ADD1:%.*]] = add i64 [[C:%.*]], [[B:%.*]]
9; CHECK-NEXT:    [[REASS_MUL2:%.*]] = mul i64 [[A:%.*]], [[A]]
10; CHECK-NEXT:    [[REASS_MUL:%.*]] = mul i64 [[REASS_MUL2]], [[REASS_ADD1]]
11; CHECK-NEXT:    ret i64 [[REASS_MUL]]
12;
13  %t0 = mul i64 %a, %b
14  %t1 = mul i64 %a, %t0 ; a*(a*b)
15  %t2 = mul i64 %a, %c
16  %t3 = mul i64 %a, %t2 ; a*(a*c)
17  %t4 = add i64 %t1, %t3
18  ret i64 %t4
19}
20
21; Check that a*b+a*c+d is turned into a*(b+c)+d.
22
23define i64 @multistep2(i64 %a, i64 %b, i64 %c, i64 %d) {
24; CHECK-LABEL: @multistep2(
25; CHECK-NEXT:    [[REASS_ADD:%.*]] = add i64 [[C:%.*]], [[B:%.*]]
26; CHECK-NEXT:    [[REASS_MUL:%.*]] = mul i64 [[REASS_ADD]], [[A:%.*]]
27; CHECK-NEXT:    [[T3:%.*]] = add i64 [[REASS_MUL]], [[D:%.*]]
28; CHECK-NEXT:    ret i64 [[T3]]
29;
30  %t0 = mul i64 %a, %b
31  %t1 = mul i64 %a, %c
32  %t2 = add i64 %t1, %d ; a*c+d
33  %t3 = add i64 %t0, %t2 ; a*b+(a*c+d)
34  ret i64 %t3
35}
36
37