1; REQUIRES: x86-registered-target
2
3; Test devirtualization through the thin link and backend.
4
5; Generate split module with summary for hybrid Thin/Regular LTO WPD.
6; RUN: opt -thinlto-bc -thinlto-split-lto-unit -o %t.o %s
7
8; Check that we have module flag showing splitting enabled, and that we don't
9; generate summary information needed for index-based WPD.
10; RUN: llvm-modextract -b -n=0 %t.o -o %t.o.0
11; RUN: llvm-dis -o - %t.o.0 | FileCheck %s --check-prefix=ENABLESPLITFLAG --implicit-check-not=vTableFuncs --implicit-check-not=typeidCompatibleVTable
12; RUN: llvm-modextract -b -n=1 %t.o -o %t.o.1
13; RUN: llvm-dis -o - %t.o.1 | FileCheck %s --check-prefix=ENABLESPLITFLAG --implicit-check-not=vTableFuncs --implicit-check-not=typeidCompatibleVTable
14; ENABLESPLITFLAG: !{i32 1, !"EnableSplitLTOUnit", i32 1}
15
16; Generate unsplit module with summary for ThinLTO index-based WPD.
17; RUN: opt -thinlto-bc -o %t2.o %s
18
19; Check that we don't have module flag when splitting not enabled for ThinLTO,
20; and that we generate summary information needed for index-based WPD.
21; RUN: llvm-dis -o - %t2.o | FileCheck %s --check-prefix=NOENABLESPLITFLAG
22; NOENABLESPLITFLAG-DAG: !{i32 1, !"EnableSplitLTOUnit", i32 0}
23; NOENABLESPLITFLAG-DAG: [[An:\^[0-9]+]] = gv: (name: "_ZN1A1nEi"
24; NOENABLESPLITFLAG-DAG: [[Bf:\^[0-9]+]] = gv: (name: "_ZN1B1fEi"
25; NOENABLESPLITFLAG-DAG: [[Cf:\^[0-9]+]] = gv: (name: "_ZN1C1fEi"
26; NOENABLESPLITFLAG-DAG: [[Dm:\^[0-9]+]] = gv: (name: "_ZN1D1mEi"
27; NOENABLESPLITFLAG-DAG: [[B:\^[0-9]+]] = gv: (name: "_ZTV1B", {{.*}} vTableFuncs: ((virtFunc: [[Bf]], offset: 16), (virtFunc: [[An]], offset: 24)), refs: ([[Bf]], [[An]])
28; NOENABLESPLITFLAG-DAG: [[C:\^[0-9]+]] = gv: (name: "_ZTV1C", {{.*}} vTableFuncs: ((virtFunc: [[Cf]], offset: 16), (virtFunc: [[An]], offset: 24)), refs: ([[An]], [[Cf]])
29; NOENABLESPLITFLAG-DAG: [[D:\^[0-9]+]] = gv: (name: "_ZTV1D", {{.*}} vTableFuncs: ((virtFunc: [[Dm]], offset: 16)), refs: ([[Dm]])
30; NOENABLESPLITFLAG-DAG: typeidCompatibleVTable: (name: "_ZTS1A", summary: ((offset: 16, [[B]]), (offset: 16, [[C]])))
31; NOENABLESPLITFLAG-DAG: typeidCompatibleVTable: (name: "_ZTS1B", summary: ((offset: 16, [[B]])))
32; NOENABLESPLITFLAG-DAG: typeidCompatibleVTable: (name: "_ZTS1C", summary: ((offset: 16, [[C]])))
33; Type Id on _ZTV1D should have been promoted
34; NOENABLESPLITFLAG-DAG: typeidCompatibleVTable: (name: "1.{{.*}}", summary: ((offset: 16, [[D]])))
35
36; Index based WPD
37; RUN: llvm-lto2 run %t2.o -save-temps -pass-remarks=. \
38; RUN:   -whole-program-visibility \
39; RUN:   -o %t3 \
40; RUN:   -r=%t2.o,test,px \
41; RUN:   -r=%t2.o,_ZN1A1nEi,p \
42; RUN:   -r=%t2.o,_ZN1B1fEi,p \
43; RUN:   -r=%t2.o,_ZN1C1fEi,p \
44; RUN:   -r=%t2.o,_ZN1D1mEi,p \
45; RUN:   -r=%t2.o,_ZTV1B,px \
46; RUN:   -r=%t2.o,_ZTV1C,px \
47; RUN:   -r=%t2.o,_ZTV1D,px 2>&1 | FileCheck %s --check-prefix=REMARK
48; RUN: llvm-dis %t3.1.4.opt.bc -o - | FileCheck %s --check-prefix=CHECK-IR
49
50; Check that we're able to prevent specific function from being
51; devirtualized when running index based WPD.
52; RUN: llvm-lto2 run %t2.o -save-temps -pass-remarks=. \
53; RUN:   -whole-program-visibility \
54; RUN:   -wholeprogramdevirt-skip=_ZN1A1nEi \
55; RUN:   -o %t3 \
56; RUN:   -r=%t2.o,test,px \
57; RUN:   -r=%t2.o,_ZN1A1nEi,p \
58; RUN:   -r=%t2.o,_ZN1B1fEi,p \
59; RUN:   -r=%t2.o,_ZN1C1fEi,p \
60; RUN:   -r=%t2.o,_ZN1D1mEi,p \
61; RUN:   -r=%t2.o,_ZTV1B,px \
62; RUN:   -r=%t2.o,_ZTV1C,px \
63; RUN:   -r=%t2.o,_ZTV1D,px 2>&1 | FileCheck %s --check-prefix=SKIP
64
65; RUN: llvm-lto2 run %t.o -save-temps -pass-remarks=. \
66; RUN:   -whole-program-visibility \
67; RUN:   -o %t3 \
68; RUN:   -r=%t.o,test,px \
69; RUN:   -r=%t.o,_ZN1A1nEi,p \
70; RUN:   -r=%t.o,_ZN1B1fEi,p \
71; RUN:   -r=%t.o,_ZN1C1fEi,p \
72; RUN:   -r=%t.o,_ZN1D1mEi,p \
73; RUN:   -r=%t.o,_ZTV1B, \
74; RUN:   -r=%t.o,_ZTV1C, \
75; RUN:   -r=%t.o,_ZTV1D, \
76; RUN:   -r=%t.o,_ZN1A1nEi, \
77; RUN:   -r=%t.o,_ZN1B1fEi, \
78; RUN:   -r=%t.o,_ZN1C1fEi, \
79; RUN:   -r=%t.o,_ZN1D1mEi, \
80; RUN:   -r=%t.o,_ZTV1B,px \
81; RUN:   -r=%t.o,_ZTV1C,px \
82; RUN:   -r=%t.o,_ZTV1D,px 2>&1 | FileCheck %s --check-prefix=REMARK --dump-input=fail
83; RUN: llvm-dis %t3.1.4.opt.bc -o - | FileCheck %s --check-prefix=CHECK-IR
84
85; REMARK-DAG: single-impl: devirtualized a call to _ZN1A1nEi
86; REMARK-DAG: single-impl: devirtualized a call to _ZN1D1mEi
87
88; SKIP-NOT: devirtualized a call to _ZN1A1nEi
89
90target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
91target triple = "x86_64-grtev4-linux-gnu"
92
93%struct.A = type { i32 (...)** }
94%struct.B = type { %struct.A }
95%struct.C = type { %struct.A }
96%struct.D = type { i32 (...)** }
97
98@_ZTV1B = constant { [4 x i8*] } { [4 x i8*] [i8* null, i8* undef, i8* bitcast (i32 (%struct.B*, i32)* @_ZN1B1fEi to i8*), i8* bitcast (i32 (%struct.A*, i32)* @_ZN1A1nEi to i8*)] }, !type !0, !type !1
99@_ZTV1C = constant { [4 x i8*] } { [4 x i8*] [i8* null, i8* undef, i8* bitcast (i32 (%struct.C*, i32)* @_ZN1C1fEi to i8*), i8* bitcast (i32 (%struct.A*, i32)* @_ZN1A1nEi to i8*)] }, !type !0, !type !2
100@_ZTV1D = constant { [3 x i8*] } { [3 x i8*] [i8* null, i8* undef, i8* bitcast (i32 (%struct.D*, i32)* @_ZN1D1mEi to i8*)] }, !type !3
101
102
103; CHECK-IR-LABEL: define i32 @test
104define i32 @test(%struct.A* %obj, %struct.D* %obj2, i32 %a) {
105entry:
106  %0 = bitcast %struct.A* %obj to i8***
107  %vtable = load i8**, i8*** %0
108  %1 = bitcast i8** %vtable to i8*
109  %p = call i1 @llvm.type.test(i8* %1, metadata !"_ZTS1A")
110  call void @llvm.assume(i1 %p)
111  %fptrptr = getelementptr i8*, i8** %vtable, i32 1
112  %2 = bitcast i8** %fptrptr to i32 (%struct.A*, i32)**
113  %fptr1 = load i32 (%struct.A*, i32)*, i32 (%struct.A*, i32)** %2, align 8
114
115  ; Check that the call was devirtualized.
116  ; CHECK-IR: %call = tail call i32 @_ZN1A1nEi
117  ; Ensure !prof and !callees metadata for indirect call promotion removed.
118  ; CHECK-IR-NOT: prof
119  ; CHECK-IR-NOT: callees
120  %call = tail call i32 %fptr1(%struct.A* nonnull %obj, i32 %a), !prof !5, !callees !6
121
122  %3 = bitcast i8** %vtable to i32 (%struct.A*, i32)**
123  %fptr22 = load i32 (%struct.A*, i32)*, i32 (%struct.A*, i32)** %3, align 8
124
125  ; We still have to call it as virtual.
126  ; CHECK-IR: %call3 = tail call i32 %fptr22
127  %call3 = tail call i32 %fptr22(%struct.A* nonnull %obj, i32 %call)
128
129  %4 = bitcast %struct.D* %obj2 to i8***
130  %vtable2 = load i8**, i8*** %4
131  %5 = bitcast i8** %vtable2 to i8*
132  %p2 = call i1 @llvm.type.test(i8* %5, metadata !4)
133  call void @llvm.assume(i1 %p2)
134
135  %6 = bitcast i8** %vtable2 to i32 (%struct.D*, i32)**
136  %fptr33 = load i32 (%struct.D*, i32)*, i32 (%struct.D*, i32)** %6, align 8
137
138  ; Check that the call was devirtualized.
139  ; CHECK-IR: %call4 = tail call i32 @_ZN1D1mEi
140  %call4 = tail call i32 %fptr33(%struct.D* nonnull %obj2, i32 %call3)
141  ret i32 %call4
142}
143; CHECK-IR-LABEL: ret i32
144; CHECK-IR-LABEL: }
145
146declare i1 @llvm.type.test(i8*, metadata)
147declare void @llvm.assume(i1)
148
149define i32 @_ZN1B1fEi(%struct.B* %this, i32 %a) #0 {
150   ret i32 0;
151}
152
153define i32 @_ZN1A1nEi(%struct.A* %this, i32 %a) #0 {
154   ret i32 0;
155}
156
157define i32 @_ZN1C1fEi(%struct.C* %this, i32 %a) #0 {
158   ret i32 0;
159}
160
161define i32 @_ZN1D1mEi(%struct.D* %this, i32 %a) #0 {
162   ret i32 0;
163}
164
165; Make sure we don't inline or otherwise optimize out the direct calls.
166attributes #0 = { noinline optnone }
167
168!0 = !{i64 16, !"_ZTS1A"}
169!1 = !{i64 16, !"_ZTS1B"}
170!2 = !{i64 16, !"_ZTS1C"}
171!3 = !{i64 16, !4}
172!4 = distinct !{}
173!5 = !{!"VP", i32 0, i64 1, i64 1621563287929432257, i64 1}
174!6 = !{i32 (%struct.A*, i32)* @_ZN1A1nEi}
175