blob: 745bcb0db8fa746222c51611ba59935706b6fd13 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
/*===- InstrProfilingExtras.c - Support library for PGO instrumentation ---===*\
|*
|* The LLVM Compiler Infrastructure
|*
|* This file is distributed under the University of Illinois Open Source
|* License. See LICENSE.TXT for details.
|*
\*===----------------------------------------------------------------------===*/
#include "InstrProfiling.h"
static void __llvm_pgo_write_file_with_name(const char *OutputName) {
FILE *OutputFile;
if (!OutputName || !OutputName[0])
return;
OutputFile = fopen(OutputName, "w");
if (!OutputFile) return;
/* TODO: mmap file to buffer of size __llvm_pgo_get_size_for_buffer() and
* pass the buffer in, instead of the file.
*/
__llvm_pgo_write_buffer(OutputFile);
fclose(OutputFile);
}
static const char *CurrentFilename = NULL;
void __llvm_pgo_set_filename(const char *Filename) {
CurrentFilename = Filename;
}
void __llvm_pgo_write_file() {
const char *Filename = CurrentFilename;
#define UPDATE_FILENAME(NextFilename) \
if (!Filename || !Filename[0]) Filename = NextFilename
UPDATE_FILENAME(getenv("LLVM_PROFILE_FILE"));
UPDATE_FILENAME("default.profdata");
#undef UPDATE_FILENAME
__llvm_pgo_write_file_with_name(Filename);
}
void __llvm_pgo_register_write_file_atexit() {
static int HasBeenRegistered = 0;
if (!HasBeenRegistered) {
HasBeenRegistered = 1;
atexit(__llvm_pgo_write_file);
}
}
|
