| // RUN: %clang_profgen -DCHECK_SYMBOLS -O3 -o %t.symbols %s |
| // RUN: llvm-nm %t.symbols | FileCheck %s --check-prefix=CHECK-SYMBOLS |
| // RUN: %clang_profgen -O3 -o %t %s |
| // RUN: %run %t %t.profraw |
| // RUN: llvm-profdata merge -o %t.profdata %t.profraw |
| // RUN: %clang_profuse=%t.profdata -o - -S -emit-llvm %s | FileCheck %s |
| |
| #include <stdint.h> |
| #include <stdlib.h> |
| |
| #ifndef CHECK_SYMBOLS |
| #include <stdio.h> |
| #endif |
| |
| int __llvm_profile_runtime = 0; |
| uint64_t __llvm_profile_get_size_for_buffer(void); |
| int __llvm_profile_write_buffer(char *); |
| void __llvm_profile_merge_from_buffer(const char *, uint64_t Size); |
| |
| int write_buffer(uint64_t, const char *); |
| int main(int argc, const char *argv[]) { |
| // CHECK-LABEL: define {{.*}} @main( |
| // CHECK: br i1 %{{.*}}, label %{{.*}}, label %{{.*}}, !prof ![[PD1:[0-9]+]] |
| if (argc < 2) |
| return 1; |
| |
| const uint64_t MaxSize = 10000; |
| static char Buffer[MaxSize]; |
| |
| uint64_t Size = __llvm_profile_get_size_for_buffer(); |
| if (Size > MaxSize) |
| return 1; |
| int Write = __llvm_profile_write_buffer(Buffer); |
| if (Write) |
| return Write; |
| |
| #ifdef CHECK_SYMBOLS |
| // Don't write it out. Since we're checking the symbols, we don't have libc |
| // available. |
| // Call merge function to make sure it does not bring in libc deps: |
| __llvm_profile_merge_from_buffer(Buffer, Size); |
| return 0; |
| #else |
| // Actually write it out so we can FileCheck the output. |
| FILE *File = fopen(argv[1], "w"); |
| if (!File) |
| return 1; |
| if (fwrite(Buffer, 1, Size, File) != Size) |
| return 1; |
| return fclose(File); |
| #endif |
| } |
| // CHECK: ![[PD1]] = !{!"branch_weights", i32 1, i32 2} |
| |
| // CHECK-SYMBOLS-NOT: {{ }}___cxx_global_var_init |
| // CHECK-SYMBOLS-NOT: {{ }}___llvm_profile_register_write_file_atexit |
| // CHECK-SYMBOLS-NOT: {{ }}___llvm_profile_set_filename |
| // CHECK-SYMBOLS-NOT: {{ }}___llvm_profile_write_file |
| // CHECK-SYMBOLS-NOT: {{ }}_fdopen |
| // CHECK-SYMBOLS-NOT: {{ }}_fopen |
| // CHECK-SYMBOLS-NOT: {{ }}_fwrite |
| // CHECK-SYMBOLS-NOT: {{ }}_getenv |
| // CHECK-SYMBOLS-NOT: {{ }}getenv |
| // CHECK-SYMBOLS-NOT: {{ }}_malloc |
| // CHECK-SYMBOLS-NOT: {{ }}malloc |
| // CHECK-SYMBOLS-NOT: {{ }}_calloc |
| // CHECK-SYMBOLS-NOT: {{ }}calloc |
| // CHECK-SYMBOLS-NOT: {{ }}_free |
| // CHECK-SYMBOLS-NOT: {{ }}free |
| // CHECK-SYMBOLS-NOT: {{ }}_open |
