accum_recursion.ll 1.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. ; RUN: opt < %s -tailcallelim -S | FileCheck %s
  2. define i32 @test1_factorial(i32 %x) {
  3. entry:
  4. %tmp.1 = icmp sgt i32 %x, 0 ; <i1> [#uses=1]
  5. br i1 %tmp.1, label %then, label %else
  6. then: ; preds = %entry
  7. %tmp.6 = add i32 %x, -1 ; <i32> [#uses=1]
  8. %tmp.4 = call i32 @test1_factorial( i32 %tmp.6 ) ; <i32> [#uses=1]
  9. %tmp.7 = mul i32 %tmp.4, %x ; <i32> [#uses=1]
  10. ret i32 %tmp.7
  11. else: ; preds = %entry
  12. ret i32 1
  13. }
  14. ; CHECK-LABEL: define i32 @test1_factorial(
  15. ; CHECK: phi i32
  16. ; CHECK-NOT: call i32
  17. ; CHECK: else:
  18. ; This is a more aggressive form of accumulator recursion insertion, which
  19. ; requires noticing that X doesn't change as we perform the tailcall.
  20. define i32 @test2_mul(i32 %x, i32 %y) {
  21. entry:
  22. %tmp.1 = icmp eq i32 %y, 0 ; <i1> [#uses=1]
  23. br i1 %tmp.1, label %return, label %endif
  24. endif: ; preds = %entry
  25. %tmp.8 = add i32 %y, -1 ; <i32> [#uses=1]
  26. %tmp.5 = call i32 @test2_mul( i32 %x, i32 %tmp.8 ) ; <i32> [#uses=1]
  27. %tmp.9 = add i32 %tmp.5, %x ; <i32> [#uses=1]
  28. ret i32 %tmp.9
  29. return: ; preds = %entry
  30. ret i32 %x
  31. }
  32. ; CHECK-LABEL: define i32 @test2_mul(
  33. ; CHECK: phi i32
  34. ; CHECK-NOT: call i32
  35. ; CHECK: return:
  36. define i64 @test3_fib(i64 %n) nounwind readnone {
  37. ; CHECK-LABEL: @test3_fib(
  38. entry:
  39. ; CHECK: tailrecurse:
  40. ; CHECK: %accumulator.tr = phi i64 [ %n, %entry ], [ %3, %bb1 ]
  41. ; CHECK: %n.tr = phi i64 [ %n, %entry ], [ %2, %bb1 ]
  42. switch i64 %n, label %bb1 [
  43. ; CHECK: switch i64 %n.tr, label %bb1 [
  44. i64 0, label %bb2
  45. i64 1, label %bb2
  46. ]
  47. bb1:
  48. ; CHECK: bb1:
  49. %0 = add i64 %n, -1
  50. ; CHECK: %0 = add i64 %n.tr, -1
  51. %1 = tail call i64 @test3_fib(i64 %0) nounwind
  52. ; CHECK: %1 = tail call i64 @test3_fib(i64 %0)
  53. %2 = add i64 %n, -2
  54. ; CHECK: %2 = add i64 %n.tr, -2
  55. %3 = tail call i64 @test3_fib(i64 %2) nounwind
  56. ; CHECK-NOT: tail call i64 @test3_fib
  57. %4 = add nsw i64 %3, %1
  58. ; CHECK: add nsw i64 %accumulator.tr, %1
  59. ret i64 %4
  60. ; CHECK: br label %tailrecurse
  61. bb2:
  62. ; CHECK: bb2:
  63. ret i64 %n
  64. ; CHECK: ret i64 %accumulator.tr
  65. }