[AIX][PGO] Enable PGO continuous mode support on AIX (#110196)

Wael Yehia · Wael Yehia · commit f0083c6af7c1 · 2024-09-27T22:34:05.000Z
diff --git a/compiler-rt/lib/profile/InstrProfilingFile.c b/compiler-rt/lib/profile/InstrProfilingFile.c
@@ -196,7 +196,7 @@ static int mmapForContinuousMode(uint64_t CurrentFileOffset, FILE *File) {
   }
   return 0;
 }
-#elif defined(__ELF__) || defined(_WIN32)
+#elif defined(__ELF__) || defined(_WIN32) || defined(_AIX)
 
 #define INSTR_PROF_PROFILE_COUNTER_BIAS_DEFAULT_VAR                            \
   INSTR_PROF_CONCAT(INSTR_PROF_PROFILE_COUNTER_BIAS_VAR, _default)
@@ -862,7 +862,7 @@ static int parseFilenamePattern(const char *FilenamePat,
           __llvm_profile_disable_continuous_mode();
           return -1;
         }
-#if defined(__APPLE__) || defined(__ELF__) || defined(_WIN32)
+#if defined(__APPLE__) || defined(__ELF__) || defined(_WIN32) || defined(_AIX)
         __llvm_profile_set_page_size(getpagesize());
         __llvm_profile_enable_continuous_mode();
 #else
diff --git a/compiler-rt/test/profile/ContinuousSyncMode/runtime-counter-relocation.c b/compiler-rt/test/profile/ContinuousSyncMode/runtime-counter-relocation.c
@@ -1,11 +1,11 @@
-// REQUIRES: target={{.*(linux|solaris|windows-msvc).*}}
+// REQUIRES: target={{.*(linux|solaris|windows-msvc|aix).*}}
 
 // RUN: %clang -fprofile-instr-generate -fcoverage-mapping -mllvm -runtime-counter-relocation=true -o %t.exe %s
 // RUN: echo "garbage" > %t.profraw
 // RUN: env LLVM_PROFILE_FILE="%c%t.profraw" %run %t.exe
 // RUN: llvm-profdata show --counts --all-functions %t.profraw | FileCheck %s -check-prefix=CHECK-COUNTS
 // RUN: llvm-profdata merge -o %t.profdata %t.profraw
-// RUN: llvm-cov report %t.exe -instr-profile %t.profdata | FileCheck %s -check-prefix=CHECK-COVERAGE
+// RUN: %if !target={{.*aix.*}} %{ llvm-cov report %t.exe -instr-profile %t.profdata | FileCheck %s -check-prefix=CHECK-COVERAGE %}
 
 // CHECK-COUNTS: Counters:
 // CHECK-COUNTS-NEXT:   main:
diff --git a/compiler-rt/test/profile/ContinuousSyncMode/set-file-object.c b/compiler-rt/test/profile/ContinuousSyncMode/set-file-object.c
@@ -1,4 +1,4 @@
-// REQUIRES: target={{.*(darwin|linux|solaris).*}}
+// REQUIRES: target={{.*(darwin|linux|solaris|aix).*}}
 
 // Test using __llvm_profile_set_file_object in continuous mode (%c).
 // Create & cd into a temporary directory.

Original file line number	Diff line number	Diff line change
`@@ -196,7 +196,7 @@ static int mmapForContinuousMode(uint64_t CurrentFileOffset, FILE *File) {`
`196`	`196`	`}`
`197`	`197`	`return 0;`
`198`	`198`	`}`
`199`		`-#elif defined(__ELF__) \|\| defined(_WIN32)`
	`199`	`+#elif defined(__ELF__) \|\| defined(_WIN32) \|\| defined(_AIX)`
`200`	`200`
`201`	`201`	`#define INSTR_PROF_PROFILE_COUNTER_BIAS_DEFAULT_VAR \`
`202`	`202`	`INSTR_PROF_CONCAT(INSTR_PROF_PROFILE_COUNTER_BIAS_VAR, _default)`
`@@ -862,7 +862,7 @@ static int parseFilenamePattern(const char *FilenamePat,`
`862`	`862`	`__llvm_profile_disable_continuous_mode();`
`863`	`863`	`return -1;`
`864`	`864`	`}`
`865`		`-#if defined(__APPLE__) \|\| defined(__ELF__) \|\| defined(_WIN32)`
	`865`	`+#if defined(__APPLE__) \|\| defined(__ELF__) \|\| defined(_WIN32) \|\| defined(_AIX)`
`866`	`866`	`__llvm_profile_set_page_size(getpagesize());`
`867`	`867`	`__llvm_profile_enable_continuous_mode();`
`868`	`868`	`#else`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-// REQUIRES: target={{.(darwin\|linux\|solaris).}}`
	`1`	`+// REQUIRES: target={{.(darwin\|linux\|solaris\|aix).}}`
`2`	`2`
`3`	`3`	`// Test using __llvm_profile_set_file_object in continuous mode (%c).`
`4`	`4`	`// Create & cd into a temporary directory.`