llvm-project/llvm/test/Transforms/FunctionSpecialization/function-specialization-recursive.ll
Alexandros Lamprineas 8136a0172b [FuncSpec] Make the Function Specializer part of the IPSCCP pass.
Reland 877a9f9abec61f06e39f1cd872e37b828139c2d1 since D138654 (parent)
has been fixed with 9ebaf4fef4aac89d4eff08e48185d61bc893f14e and with
8f1e11c5a7d70f96943a72649daa69f152d73e90.

Differential Revision: https://reviews.llvm.org/D126455
2022-12-10 14:39:49 +00:00

60 lines
2.4 KiB
LLVM

; RUN: opt -passes=ipsccp,inline,instcombine -specialize-functions -force-function-specialization -func-specialization-max-iters=2 -S < %s | FileCheck %s --check-prefix=ITERS2
; RUN: opt -passes=ipsccp,inline,instcombine -specialize-functions -force-function-specialization -func-specialization-max-iters=3 -S < %s | FileCheck %s --check-prefix=ITERS3
; RUN: opt -passes=ipsccp,inline,instcombine -specialize-functions -force-function-specialization -func-specialization-max-iters=4 -S < %s | FileCheck %s --check-prefix=ITERS4
@low = internal constant i32 0, align 4
@high = internal constant i32 6, align 4
define internal void @recursiveFunc(ptr nocapture readonly %lo, i32 %step, ptr nocapture readonly %hi) {
%lo.temp = alloca i32, align 4
%hi.temp = alloca i32, align 4
%lo.load = load i32, ptr %lo, align 4
%hi.load = load i32, ptr %hi, align 4
%cmp = icmp ne i32 %lo.load, %hi.load
br i1 %cmp, label %block6, label %ret.block
block6:
call void @print_val(i32 %lo.load, i32 %hi.load)
%add = add nsw i32 %lo.load, %step
%sub = sub nsw i32 %hi.load, %step
store i32 %add, ptr %lo.temp, align 4
store i32 %sub, ptr %hi.temp, align 4
call void @recursiveFunc(ptr nonnull %lo.temp, i32 %step, ptr nonnull %hi.temp)
br label %ret.block
ret.block:
ret void
}
; ITERS2: @funcspec.arg.4 = internal constant i32 2
; ITERS2: @funcspec.arg.5 = internal constant i32 4
; ITERS3: @funcspec.arg.7 = internal constant i32 3
; ITERS3: @funcspec.arg.8 = internal constant i32 3
define i32 @main() {
; ITERS2-LABEL: @main(
; ITERS2-NEXT: call void @print_val(i32 0, i32 6)
; ITERS2-NEXT: call void @print_val(i32 1, i32 5)
; ITERS2-NEXT: call void @recursiveFunc(ptr nonnull @funcspec.arg.4, i32 1, ptr nonnull @funcspec.arg.5)
; ITERS2-NEXT: ret i32 0
;
; ITERS3-LABEL: @main(
; ITERS3-NEXT: call void @print_val(i32 0, i32 6)
; ITERS3-NEXT: call void @print_val(i32 1, i32 5)
; ITERS3-NEXT: call void @print_val(i32 2, i32 4)
; ITERS3-NEXT: call void @recursiveFunc(ptr nonnull @funcspec.arg.7, i32 1, ptr nonnull @funcspec.arg.8)
; ITERS3-NEXT: ret i32 0
;
; ITERS4-LABEL: @main(
; ITERS4-NEXT: call void @print_val(i32 0, i32 6)
; ITERS4-NEXT: call void @print_val(i32 1, i32 5)
; ITERS4-NEXT: call void @print_val(i32 2, i32 4)
; ITERS4-NEXT: ret i32 0
;
call void @recursiveFunc(ptr nonnull @low, i32 1, ptr nonnull @high)
ret i32 0
}
declare dso_local void @print_val(i32, i32)