-
Notifications
You must be signed in to change notification settings - Fork 10.9k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[FuncSpec] Relax restrictions on candidates for specialisation
Allow a function to be specialised even if it has its address taken or it's global. For such functions, consider all of the arguments as overdefined. Don't delete the functions even if all the apparent calls were redirected to specialised instances. Reviewed By: labrinea, ChuanqiXu Differential Revision: https://reviews.llvm.org/D148345
- Loading branch information
1 parent
99d4c72
commit cc7bb70
Showing
2 changed files
with
74 additions
and
5 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
64 changes: 64 additions & 0 deletions
64
llvm/test/Transforms/FunctionSpecialization/non-argument-tracked.ll
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,64 @@ | ||
; RUN: opt -S --passes=ipsccp,deadargelim -funcspec-for-literal-constant --force-specialization < %s | FileCheck %s | ||
|
||
; Test that all of `f0`, `f1`, and `f2` are specialised, even though `f0` has its address taken | ||
; and `f1` is with external linkage (`f2` was specialised anyway). | ||
|
||
@p = global ptr @f0 | ||
|
||
; `f0` is kept even though all apparent calls are specialized | ||
; CHECK-LABEL: define internal i32 @f0( | ||
define internal i32 @f0(i32 %i) { | ||
%v = add i32 %i, 1 | ||
ret i32 %v | ||
} | ||
|
||
; Likewise, `f1` is kept, because of the external linkage | ||
; CHECK-LABEL: define i32 @f1( | ||
define i32 @f1(i32 %i) { | ||
%v = add i32 %i, 1 | ||
ret i32 %v | ||
} | ||
|
||
; `f2` is fully specialised. | ||
; CHECK-NOT: defined internal i32 @f2() | ||
define internal i32 @f2(i32 %i) { | ||
%v = add i32 %i, 1 | ||
ret i32 %v | ||
} | ||
|
||
;; All calls are to specilisation instances. | ||
|
||
; CHECK-LABEL: define i32 @g0 | ||
; CHECK: [[U0:%.*]] = call i32 @f0.[[#A:]]() | ||
; CHECK-NEXT: [[U1:%.*]] = call i32 @f1.[[#B:]]() | ||
; CHECK-NEXT: [[U2:%.*]] = call i32 @f2.[[#C:]]() | ||
define i32 @g0(i32 %i) { | ||
%u0 = call i32 @f0(i32 1) | ||
%u1 = call i32 @f1(i32 2) | ||
%u2 = call i32 @f2(i32 3) | ||
%v0 = add i32 %u0, %u1 | ||
%v = add i32 %v0, %u2 | ||
ret i32 %v | ||
} | ||
|
||
; CHECK-LABEL: define i32 @g1 | ||
; CHECK: [[U0:%.*]] = call i32 @f0.[[#D:]]() | ||
; CHECK-NEXT: [[U1:%.*]] = call i32 @f1.[[#E:]]() | ||
; CHECK-NEXT: [[U2:%.*]] = call i32 @f2.[[#F:]]() | ||
define i32 @g1(i32 %i) { | ||
%u0 = call i32 @f0(i32 2) | ||
%u1 = call i32 @f1(i32 3) | ||
%u2 = call i32 @f2(i32 4) | ||
%v0 = add i32 %u0, %u1 | ||
%v = add i32 %v0, %u2 | ||
ret i32 %v | ||
} | ||
|
||
; All of the function are specialized and all clones are with internal linkage. | ||
|
||
; CHECK-DAG: define internal i32 @f0.[[#A]]() { | ||
; CHECK-DAG: define internal i32 @f1.[[#B]]() { | ||
; CHECK-DAG: define internal i32 @f2.[[#C]]() { | ||
; CHECK-DAG: define internal i32 @f0.[[#D]]() { | ||
; CHECK-DAG: define internal i32 @f1.[[#E]]() { | ||
; CHECK-DAG: define internal i32 @f2.[[#F]]() { |