test/Transforms/NaryReassociate/NVPTX/nary-slsr.ll

   1 ; RUN: opt < %s -slsr -nary-reassociate -S | FileCheck %s
   2 ; RUN: llc < %s -march=nvptx64 -mcpu=sm_20 | FileCheck %s --check-prefix=PTX
   3
   4 target datalayout = "e-i64:64-v16:16-v32:32-n16:32:64"
   5
   6 ; foo((a + b) + c);
   7 ; foo((a + b * 2) + c);
   8 ; foo((a + b * 3) + c);
   9 ;   =>
  10 ; abc = (a + b) + c;
  11 ; foo(abc);
  12 ; ab2c = abc + b;
  13 ; foo(ab2c);
  14 ; ab3c = ab2c + b;
  15 ; foo(ab3c);
  16 define void @nary_reassociate_after_slsr(i32 %a, i32 %b, i32 %c) {
  17 ; CHECK-LABEL: @nary_reassociate_after_slsr(
  18 ; PTX-LABEL: .visible .func nary_reassociate_after_slsr(
  19 ; PTX: ld.param.u32 [[b:%r[0-9]+]], [nary_reassociate_after_slsr_param_1];
  20   %ab = add i32 %a, %b
  21   %abc = add i32 %ab, %c
  22   call void @foo(i32 %abc)
  23 ; CHECK: call void @foo(i32 %abc)
  24 ; PTX: st.param.b32 [param0+0], [[abc:%r[0-9]+]];
  25
  26   %b2 = shl i32 %b, 1
  27   %ab2 = add i32 %a, %b2
  28   %ab2c = add i32 %ab2, %c
  29 ; CHECK-NEXT: %ab2c = add i32 %abc, %b
  30 ; PTX: add.s32 [[ab2c:%r[0-9]+]], [[abc]], [[b]]
  31   call void @foo(i32 %ab2c)
  32 ; CHECK-NEXT: call void @foo(i32 %ab2c)
  33 ; PTX: st.param.b32 [param0+0], [[ab2c]];
  34
  35   %b3 = mul i32 %b, 3
  36   %ab3 = add i32 %a, %b3
  37   %ab3c = add i32 %ab3, %c
  38 ; CHECK-NEXT: %ab3c = add i32 %ab2c, %b
  39 ; PTX: add.s32 [[ab3c:%r[0-9]+]], [[ab2c]], [[b]]
  40   call void @foo(i32 %ab3c)
  41 ; CHECK-NEXT: call void @foo(i32 %ab3c)
  42 ; PTX: st.param.b32 [param0+0], [[ab3c]];
  43
  44   ret void
  45 }
  46
  47 declare void @foo(i32)