test/Transforms/Reassociate/multistep.ll

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