237 lines · plain
1; RUN: opt -S %s -passes=sample-profile -sample-profile-file=%S/Inputs/indirect-call.prof | FileCheck %s2; RUN: opt -S %s -passes=sample-profile -sample-profile-file=%S/Inputs/indirect-call.extbinary.afdo | FileCheck %s3 4; CHECK-LABEL: @test5define void @test(ptr) #0 !dbg !3 {6 %2 = alloca ptr7 store ptr %0, ptr %28 %3 = load ptr, ptr %29 ; CHECK: call {{.*}}, !prof ![[PROF:[0-9]+]]10 call void %3(), !dbg !411 ret void12}13 14; CHECK-LABEL: @test_inline15; If the indirect call is promoted and inlined in profile, we should promote and inline it.16define void @test_inline(ptr, ptr %x) #0 !dbg !6 {17 %2 = alloca ptr18 store ptr %0, ptr %219 %3 = load ptr, ptr %220; CHECK: icmp {{.*}} @foo_inline221; CHECK: br {{.*}} !prof ![[BR1:[0-9]+]]22; CHECK: if.true.direct_targ:23; CHECK-NOT: call24; CHECK: if.false.orig_indirect:25; CHECK: icmp {{.*}} @foo_inline126; CHECK: br {{.*}} !prof ![[BR2:[0-9]+]]27; CHECK: if.true.direct_targ1:28; CHECK-NOT: call29; CHECK: if.false.orig_indirect2:30; CHECK: call {{.*}} !prof ![[VP:[0-9]+]]31 call ptr %3(ptr %x), !dbg !732 ret void33}34 35; CHECK-LABEL: @test_inline_strip36; If the indirect call is promoted and inlined in profile, and the callee name37; is stripped we should promote and inline it.38define void @test_inline_strip(ptr, ptr %x) #0 !dbg !8 {39 %2 = alloca ptr40 store ptr %0, ptr %241 %3 = load ptr, ptr %242; CHECK: icmp {{.*}} @foo_inline_strip.suffix43; CHECK: if.true.direct_targ:44; CHECK-NOT: call45; CHECK: if.false.orig_indirect:46; CHECK: call47 call ptr %3(ptr %x), !dbg !948 ret void49}50 51; CHECK-LABEL: @test_inline_strip_conflict52; If the indirect call is promoted and inlined in profile, and the callee name53; is stripped, but have more than 1 potential match, we should not promote.54define void @test_inline_strip_conflict(ptr, ptr %x) #0 !dbg !10 {55 %2 = alloca ptr56 store ptr %0, ptr %257 %3 = load ptr, ptr %258; CHECK-NOT: if.true.direct_targ:59 call ptr %3(ptr %x), !dbg !1160 ret void61}62 63; CHECK-LABEL: @test_noinline64; If the indirect call target is not available, we should not promote it.65define void @test_noinline(ptr) #0 !dbg !12 {66 %2 = alloca ptr67 store ptr %0, ptr %268 %3 = load ptr, ptr %269; CHECK-NOT: icmp70; CHECK: call71 call void %3(), !dbg !1372 ret void73}74 75; CHECK-LABEL: @test_noinline_bitcast76; If the indirect call has been promoted to a direct call with bitcast,77; do not inline it.78define float @test_noinline_bitcast(ptr) #0 !dbg !26 {79 %2 = alloca ptr80 store ptr %0, ptr %281; CHECK: icmp82; CHECK: call83 %3 = load ptr, ptr %284 %4 = call float %3(), !dbg !2785 ret float %486}87 88; CHECK-LABEL: @test_norecursive_inline89; If the indirect call target is the caller, we should not promote it.90define void @test_norecursive_inline() #0 !dbg !24 {91; CHECK-NOT: icmp92; CHECK: call93 %1 = load ptr, ptr @y, align 894 call void %1(), !dbg !2595 ret void96}97 98define ptr @return_arg(ptr readnone returned) #0 !dbg !29{99 ret ptr %0100}101 102; CHECK-LABEL: @return_arg_caller103; When the promoted indirect call returns a parameter that was defined by the104; return value of a previous direct call. Checks both direct call and promoted105; indirect call are inlined.106define ptr @return_arg_caller(ptr nocapture) #0 !dbg !30{107; CHECK-NOT: call ptr @foo_inline1108; CHECK: if.true.direct_targ:109; CHECK-NOT: call110; CHECK: if.false.orig_indirect:111; CHECK: call112 %2 = call ptr @foo_inline1(ptr null), !dbg !31113 %cmp = icmp ne ptr %2, null114 br i1 %cmp, label %then, label %else115 116then:117 %3 = tail call ptr %0(ptr %2), !dbg !32118 ret ptr %3119 120else:121 ret ptr null122}123 124; CHECK-LABEL: @branch_prof_valid125; Check the conditional branch generated by indirect call promotion won't126; have invalid profile like !{!"branch_weights", i32 0, i32 0}.127define void @branch_prof_valid(ptr %t0) #0 !dbg !33 {128 %t1 = alloca ptr129 store ptr %t0, ptr %t1130 %t2 = load ptr, ptr %t1131 ; CHECK-NOT: call {{.*}}132 ; CHECK: br i1 {{.*}}, label %if.true.direct_targ, label %if.false.orig_indirect, {{.*}}, !prof ![[BR3:[0-9]+]]133 call void %t2(), !dbg !34134 ret void135}136 137@x = global i32 0, align 4138@y = global ptr null, align 8139 140define ptr @foo_inline1(ptr %x) #0 !dbg !14 {141 ret ptr %x142}143 144define ptr @foo_inline_strip.suffix(ptr %x) #0 !dbg !15 {145 ret ptr %x146}147 148define ptr @foo_inline_strip_conflict.suffix1(ptr %x) #0 !dbg !16 {149 ret ptr %x150}151 152define ptr @foo_inline_strip_conflict.suffix2(ptr %x) #0 !dbg !17 {153 ret ptr %x154}155 156define ptr @foo_inline_strip_conflict.suffix3(ptr %x) #0 !dbg !18 {157 ret ptr %x158}159 160define ptr @foo_inline2(ptr %x) #0 !dbg !19 {161 ret ptr %x162}163 164define void @foo_inline3() #0 !dbg !35 {165 ret void166}167 168define i32 @foo_noinline(i32 %x) #0 !dbg !20 {169 ret i32 %x170}171 172define void @foo_direct() #0 !dbg !21 {173 ret void174}175 176define i32 @foo_direct_i32() #0 !dbg !28 {177 ret i32 0;178}179 180; CHECK-LABEL: @test_direct181; We should not promote a direct call.182define void @test_direct() #0 !dbg !22 {183; CHECK-NOT: icmp184; CHECK: call185 call void @foo_alias(), !dbg !23186 ret void187}188 189@foo_alias = alias void (), ptr @foo_direct190 191attributes #0 = {"use-sample-profile"}192 193!llvm.dbg.cu = !{!0}194!llvm.module.flags = !{!2}195 196!0 = distinct !DICompileUnit(language: DW_LANG_C_plus_plus, file: !1)197!1 = !DIFile(filename: "test.cc", directory: "/")198!2 = !{i32 2, !"Debug Info Version", i32 3}199!3 = distinct !DISubprogram(name: "test", scope: !1, file: !1, line: 3, unit: !0)200!4 = !DILocation(line: 4, scope: !3)201!5 = !DILocation(line: 6, scope: !3)202; CHECK: ![[PROF]] = !{!"VP", i32 0, i64 3457, i64 9191153033785521275, i64 2059, i64 -1069303473483922844, i64 1398}203; CHECK: ![[BR1]] = !{!"branch_weights", i32 4000, i32 4000}204; CHECK: ![[BR2]] = !{!"branch_weights", i32 3000, i32 1000}205; CHECK: ![[VP]] = !{!"VP", i32 0, i64 8000, i64 7476224446746900038, i64 -1, i64 925324185419832389, i64 -1, i64 -6391416044382067764, i64 1000}206; CHECK: ![[BR3]] = !{!"branch_weights", i32 1, i32 0}207!6 = distinct !DISubprogram(name: "test_inline", scope: !1, file: !1, line: 6, unit: !0)208!7 = !DILocation(line: 7, scope: !6)209!8 = distinct !DISubprogram(name: "test_inline_strip", scope: !1, file: !1, line: 8, unit: !0)210!9 = !DILocation(line: 9, scope: !8)211!10 = distinct !DISubprogram(name: "test_inline_strip_conflict", scope: !1, file: !1, line: 10, unit: !0)212!11 = !DILocation(line: 11, scope: !10)213!12 = distinct !DISubprogram(name: "test_noinline", scope: !1, file: !1, line: 12, unit: !0)214!13 = !DILocation(line: 13, scope: !12)215!14 = distinct !DISubprogram(name: "foo_inline1", scope: !1, file: !1, line: 11, unit: !0)216!15 = distinct !DISubprogram(name: "foo_inline_strip.suffix", scope: !1, file: !1, line: 1, unit: !0)217!16 = distinct !DISubprogram(name: "foo_inline_strip_conflict.suffix1", scope: !1, file: !1, line: 1, unit: !0)218!17 = distinct !DISubprogram(name: "foo_inline_strip_conflict.suffix2", scope: !1, file: !1, line: 1, unit: !0)219!18 = distinct !DISubprogram(name: "foo_inline_strip_conflict.suffix3", scope: !1, file: !1, line: 1, unit: !0)220!19 = distinct !DISubprogram(name: "foo_inline2", scope: !1, file: !1, line: 19, unit: !0)221!20 = distinct !DISubprogram(name: "foo_noinline", scope: !1, file: !1, line: 20, unit: !0)222!21 = distinct !DISubprogram(name: "foo_direct", scope: !1, file: !1, line: 21, unit: !0)223!22 = distinct !DISubprogram(name: "test_direct", scope: !1, file: !1, line: 22, unit: !0)224!23 = !DILocation(line: 23, scope: !22)225!24 = distinct !DISubprogram(name: "test_norecursive_inline", scope: !1, file: !1, line: 12, unit: !0)226!25 = !DILocation(line: 13, scope: !24)227!26 = distinct !DISubprogram(name: "test_noinline_bitcast", scope: !1, file: !1, line: 12, unit: !0)228!27 = !DILocation(line: 13, scope: !26)229!28 = distinct !DISubprogram(name: "foo_direct_i32", scope: !1, file: !1, line: 11, unit: !0)230!29 = distinct !DISubprogram(name: "return_arg", scope: !1, file: !1, line: 11, unit: !0)231!30 = distinct !DISubprogram(name: "return_arg_caller", scope: !1, file: !1, line: 11, unit: !0)232!31 = !DILocation(line: 12, scope: !30)233!32 = !DILocation(line: 13, scope: !30)234!33 = distinct !DISubprogram(name: "branch_prof_valid", scope: !1, file: !1, line: 25, unit: !0)235!34 = !DILocation(line: 27, scope: !33)236!35 = distinct !DISubprogram(name: "foo_inline3", scope: !1, file: !1, line: 29, unit: !0)237