[compiler-rt] [AIX][PGO] Add testcase for D136192 (PR #133758)

[Wael Yehia via llvm-commits]

https://github.com/w2yehia created https://github.com/llvm/llvm-project/pull/133758

Differential Revision: https://reviews.llvm.org/D136192

>From d1f951561dc1ec1d639d2208cfe68c4fd6b55088 Mon Sep 17 00:00:00 2001
From: Wael Yehia <wyehia at ca.ibm.com>
Date: Mon, 31 Mar 2025 17:33:57 +0000
Subject: [PATCH] [AIX][PGO] Add testcase for D136192

Differential Revision: https://reviews.llvm.org/D136192
---
 .../AIX/pgo-lto-bcdtor-function-section.test  | 119 ++++++++++++++++++
 1 file changed, 119 insertions(+)
 create mode 100755 compiler-rt/test/profile/AIX/pgo-lto-bcdtor-function-section.test

diff --git a/compiler-rt/test/profile/AIX/pgo-lto-bcdtor-function-section.test b/compiler-rt/test/profile/AIX/pgo-lto-bcdtor-function-section.test
new file mode 100755
index 0000000000000..36170716b56c2
--- /dev/null
+++ b/compiler-rt/test/profile/AIX/pgo-lto-bcdtor-function-section.test
@@ -0,0 +1,119 @@
+// RUN: split-file %s %t
+// RUN: cd %t
+//
+
+//--- foo.c
+int foo() { return 3; }
+
+//--- main.c
+int foo();
+int main() { return foo() - 3; }
+
+// # no LTO
+// ## PGO, with and without function-sections, and all permutations of -bcdtors
+// RUN: %clang_pgogen -O2 -c -fno-function-sections foo.c
+// RUN: %clang_pgogen -O2 -c -fno-function-sections main.c
+//
+// RUN: %clang_pgogen -Wl,-bcdtors:all foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=BOTH
+//
+// RUN: %clang_pgogen -Wl,-bcdtors:mbr foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=BOTH
+
+// RUN: %clang_pgogen -O2 -c -ffunction-sections foo.c
+// RUN: %clang_pgogen -O2 -c -ffunction-sections main.c
+//
+// RUN: %clang_pgogen -Wl,-bcdtors:all foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=BOTH
+//
+// RUN: %clang_pgogen -Wl,-bcdtors:mbr foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=BOTH
+
+// ## no PGO at compile step, but PGO at link step.
+// RUN: %clang -O2 -c  foo.c
+// RUN: %clang -O2 -c  main.c
+//
+// RUN: %clang_pgogen -Wl,-bcdtors:all foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=NONE
+//
+// RUN: %clang_pgogen -Wl,-bcdtors:mbr foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=NONE
+
+// # LTO, with and without function-sections, and all permutations of -bcdtors
+// ## LTO one file, no PGO at compile, PGO at link
+// RUN: %clang -O2 -c  foo.c
+// RUN: %clang -O2 -c -flto main.c
+//
+// RUN: %clang_pgogen -flto -fno-function-sections -Wl,-bcdtors:all foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=NONE
+//
+// RUN: %clang_pgogen -flto -fno-function-sections -Wl,-bcdtors:mbr foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=NONE
+//
+// RUN: %clang_pgogen -flto -ffunction-sections -Wl,-bcdtors:all foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=NONE
+//
+// RUN: %clang_pgogen -flto -ffunction-sections -Wl,-bcdtors:mbr foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=NONE
+
+// ## LTO one file, PGO at compile and link
+// RUN: %clang -O2 -c -fno-function-sections foo.c
+// RUN: %clang_pgogen -O2 -c -flto main.c
+// RUN: %clang_pgogen -flto -fno-function-sections -Wl,-bcdtors:all foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=MAIN
+
+// RUN: %clang -O2 -c -flto foo.c
+// RUN: %clang_pgogen -O2 -c -fno-function-sections main.c
+// RUN: %clang_pgogen -flto -fno-function-sections -Wl,-bcdtors:mbr foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=MAIN
+
+// RUN: %clang -O2 -c -flto foo.c
+// RUN: %clang_pgogen -O2 -c -ffunction-sections main.c
+// RUN: %clang_pgogen -flto -ffunction-sections -Wl,-bcdtors:all foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=MAIN
+
+// RUN: %clang -O2 -c -ffunction-sections foo.c
+// RUN: %clang_pgogen -O2 -c -flto main.c
+// RUN: %clang_pgogen -flto -ffunction-sections -Wl,-bcdtors:mbr foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=MAIN
+
+// ## LTO and PGO both files
+// RUN: %clang_pgogen -O2 -c -flto foo.c
+// RUN: %clang_pgogen -O2 -c -flto main.c
+//
+// RUN: %clang_pgogen -flto -fno-function-sections -Wl,-bcdtors:all foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=BOTH
+//
+// RUN: %clang_pgogen -flto -fno-function-sections -Wl,-bcdtors:mbr foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=BOTH
+//
+// RUN: %clang_pgogen -flto -ffunction-sections -Wl,-bcdtors:all foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=BOTH
+//
+// RUN: %clang_pgogen -flto -ffunction-sections -Wl,-bcdtors:mbr foo.o main.o
+// RUN: rm -f default* && %run a.out
+// RUN: llvm-profdata show --all-functions default* | FileCheck  %s -check-prefix=BOTH
+
+// BOTH-DAG: foo:
+// BOTH-DAG: main:
+// MAIN-NOT: foo:
+// MAIN: main:
+// MAIN-NOT: foo:
+// NONE: Total functions: 0