1 // Tests for instrumentation of C++11 lambdas
2 
3 // RUN: %clang_cc1 -x c++ %s -triple %itanium_abi_triple -main-file-name cxx-lambda.cpp -std=c++11 -o - -emit-llvm -fprofile-instr-generate > %tgen
4 // RUN: FileCheck --input-file=%tgen -check-prefix=PGOGEN %s
5 // RUN: FileCheck --input-file=%tgen -check-prefix=LMBGEN %s
6 
7 // RUN: %clang_cc1 -x c++ %s -triple %itanium_abi_triple -main-file-name cxx-lambda.cpp -std=c++11 -o - -emit-llvm -fprofile-instr-use=%S/Inputs/cxx-lambda.profdata > %tuse
8 // RUN: FileCheck --input-file=%tuse -check-prefix=PGOUSE %s
9 // RUN: FileCheck --input-file=%tuse -check-prefix=LMBUSE %s
10 
11 // PGOGEN: @[[LWC:__llvm_profile_counters__Z7lambdasv]] = global [4 x i64] zeroinitializer
12 // PGOGEN: @[[MAC:__llvm_profile_counters_main]] = global [1 x i64] zeroinitializer
13 // LMBGEN: @[[LFC:"__llvm_profile_counters__ZZ7lambdasvENK3\$_0clEi"]] = internal global [3 x i64] zeroinitializer
14 
15 // PGOGEN-LABEL: define void @_Z7lambdasv()
16 // PGOUSE-LABEL: define void @_Z7lambdasv()
17 // PGOGEN: store {{.*}} @[[LWC]], i64 0, i64 0
18 void lambdas() {
19   int i = 1;
20 
21   // LMBGEN-LABEL: define internal zeroext i1 @"_ZZ7lambdasvENK3$_0clEi"(
22   // LMBUSE-LABEL: define internal zeroext i1 @"_ZZ7lambdasvENK3$_0clEi"(
23   // LMBGEN: store {{.*}} @[[LFC]], i64 0, i64 0
24   auto f = [&i](int k) {
25     // LMBGEN: store {{.*}} @[[LFC]], i64 0, i64 1
26     // LMBUSE: br {{.*}} !prof ![[LF1:[0-9]+]]
27     if (i > 0) {}
28     // LMBGEN: store {{.*}} @[[LFC]], i64 0, i64 2
29     // LMBUSE: br {{.*}} !prof ![[LF2:[0-9]+]]
30     return k && i;
31   };
32 
33   // PGOGEN: store {{.*}} @[[LWC]], i64 0, i64 1
34   // PGOUSE: br {{.*}} !prof ![[LW1:[0-9]+]]
35   if (i) {}
36 
37   // PGOGEN: store {{.*}} @[[LWC]], i64 0, i64 2
38   // PGOUSE: br {{.*}} !prof ![[LW2:[0-9]+]]
39   for (i = 0; i < 10; ++i)
40     f(9 - i);
41 
42   // PGOGEN: store {{.*}} @[[LWC]], i64 0, i64 3
43   // PGOUSE: br {{.*}} !prof ![[LW3:[0-9]+]]
44   if (i) {}
45 }
46 
47 // PGOUSE-DAG: ![[LW1]] = metadata !{metadata !"branch_weights", i32 2, i32 1}
48 // PGOUSE-DAG: ![[LW2]] = metadata !{metadata !"branch_weights", i32 11, i32 2}
49 // PGOUSE-DAG: ![[LW3]] = metadata !{metadata !"branch_weights", i32 2, i32 1}
50 
51 // LMBUSE-DAG: ![[LF1]] = metadata !{metadata !"branch_weights", i32 10, i32 2}
52 // LMBUSE-DAG: ![[LF2]] = metadata !{metadata !"branch_weights", i32 10, i32 2}
53 
54 int main(int argc, const char *argv[]) {
55   lambdas();
56   return 0;
57 }
58