1 // RUN: %clang_profgen -DCHECK_SYMBOLS -O3 -o %t.symbols %s 2 // RUN: llvm-nm %t.symbols | FileCheck %s --check-prefix=CHECK-SYMBOLS 3 // RUN: %clang_profgen -O3 -o %t %s 4 // RUN: %run %t %t.profraw 5 // RUN: llvm-profdata merge -o %t.profdata %t.profraw 6 // RUN: %clang_profuse=%t.profdata -o - -S -emit-llvm %s | FileCheck %s 7 8 #include <stdint.h> 9 #include <stdlib.h> 10 11 #ifndef CHECK_SYMBOLS 12 #include <stdio.h> 13 #endif 14 15 int __llvm_profile_runtime = 0; 16 uint64_t __llvm_profile_get_size_for_buffer(void); 17 int __llvm_profile_write_buffer(char *); 18 void __llvm_profile_merge_from_buffer(const char *, uint64_t Size); 19 20 int write_buffer(uint64_t, const char *); 21 int main(int argc, const char *argv[]) { 22 // CHECK-LABEL: define {{.*}} @main( 23 // CHECK: br i1 %{{.*}}, label %{{.*}}, label %{{.*}}, !prof ![[PD1:[0-9]+]] 24 if (argc < 2) 25 return 1; 26 27 const uint64_t MaxSize = 10000; 28 static char Buffer[MaxSize]; 29 30 uint64_t Size = __llvm_profile_get_size_for_buffer(); 31 if (Size > MaxSize) 32 return 1; 33 int Write = __llvm_profile_write_buffer(Buffer); 34 if (Write) 35 return Write; 36 37 #ifdef CHECK_SYMBOLS 38 // Don't write it out. Since we're checking the symbols, we don't have libc 39 // available. 40 // Call merge function to make sure it does not bring in libc deps: 41 __llvm_profile_merge_from_buffer(Buffer, Size); 42 return 0; 43 #else 44 // Actually write it out so we can FileCheck the output. 45 FILE *File = fopen(argv[1], "w"); 46 if (!File) 47 return 1; 48 if (fwrite(Buffer, 1, Size, File) != Size) 49 return 1; 50 return fclose(File); 51 #endif 52 } 53 // CHECK: ![[PD1]] = !{!"branch_weights", i32 1, i32 2} 54 55 // CHECK-SYMBOLS-NOT: {{ }}___cxx_global_var_init 56 // CHECK-SYMBOLS-NOT: {{ }}___llvm_profile_register_write_file_atexit 57 // CHECK-SYMBOLS-NOT: {{ }}___llvm_profile_set_filename 58 // CHECK-SYMBOLS-NOT: {{ }}___llvm_profile_write_file 59 // CHECK-SYMBOLS-NOT: {{ }}_fdopen 60 // CHECK-SYMBOLS-NOT: {{ }}_fopen 61 // CHECK-SYMBOLS-NOT: {{ }}_fwrite 62 // CHECK-SYMBOLS-NOT: {{ }}_getenv 63 // CHECK-SYMBOLS-NOT: {{ }}getenv 64 // CHECK-SYMBOLS-NOT: {{ }}_malloc 65 // CHECK-SYMBOLS-NOT: {{ }}malloc 66 // CHECK-SYMBOLS-NOT: {{ }}_calloc 67 // CHECK-SYMBOLS-NOT: {{ }}calloc 68 // CHECK-SYMBOLS-NOT: {{ }}_free 69 // CHECK-SYMBOLS-NOT: {{ }}free 70 // CHECK-SYMBOLS-NOT: {{ }}_open 71