llvm-project
93 строки · 2.3 Кб
1// RUN: %clang_pgogen -O2 -mllvm -pgo-function-entry-coverage -o %t %s
2// RUN: %run %t %t.profraw 1 1
3// RUN: llvm-profdata show --all-functions --counts %t.profraw | FileCheck %s
4
5// FIXME: llvm-profdata exits with "Malformed instrumentation profile data"
6// XFAIL: target={{.*windows.*}}
7
8#include "profile_test.h"9#include <stdint.h>10#include <stdio.h>11#include <stdlib.h>12
13int __llvm_profile_runtime = 0;14uint64_t __llvm_profile_get_size_for_buffer(void);15int __llvm_profile_write_buffer(char *);16void __llvm_profile_reset_counters(void);17int __llvm_profile_merge_from_buffer(const char *, uint64_t);18
19__attribute__((noinline)) int dumpBuffer(const char *FileN, const char *Buffer,20uint64_t Size) {21FILE *File = fopen(FileN, "w");22if (!File)23return 1;24if (fwrite(Buffer, 1, Size, File) != Size)25return 1;26return fclose(File);27}
28
29int g = 0;30__attribute__((noinline)) void foo(char c) {31if (c == '1')32g++;33else34g--;35}
36
37/* This function is not profiled */
38__attribute__((noinline)) void bar(int M) { g += M; }39
40int main(int argc, const char *argv[]) {41int i;42if (argc < 4)43return 1;44
45const uint64_t MaxSize = 10000;46static ALIGNED(sizeof(uint64_t)) char Buffer[MaxSize];47
48uint64_t Size = __llvm_profile_get_size_for_buffer();49if (Size > MaxSize)50return 1;51
52/* Start profiling. */53__llvm_profile_reset_counters();54foo(argv[2][0]);55/* End profiling by freezing counters. */56if (__llvm_profile_write_buffer(Buffer))57return 1;58
59/* Its profile will be discarded. */60for (i = 0; i < 10; i++)61bar(1);62
63/* Start profiling again and merge in previously64saved counters in buffer. */
65__llvm_profile_reset_counters();66__llvm_profile_merge_from_buffer(Buffer, Size);67foo(argv[3][0]);68/* End profiling */69if (__llvm_profile_write_buffer(Buffer))70return 1;71
72/* Its profile will be discarded. */73bar(2);74
75/* Now it is time to dump the profile to file. */76return dumpBuffer(argv[1], Buffer, Size);77}
78
79// CHECK-LABEL: dumpBuffer:
80// CHECK: Counters: 1
81// CHECK-NEXT: Block counts: [0]
82
83// CHECK-LABEL: foo:
84// CHECK: Counters: 1
85// CHECK-NEXT: Block counts: [1]
86
87// CHECK-LABEL: bar:
88// CHECK: Counters: 1
89// CHECK-NEXT: Block counts: [0]
90
91// CHECK-LABEL: main:
92// CHECK: Counters: 1
93// CHECK-NEXT: Block counts: [0]
94