Skip to content

Commit 37b0cb6

Browse files
committed
unix: BOLT fixes
As part of investigating failures with BOLT when upgrading to LLVM 19, I found and fixed a few issues with BOLT. First, `test_embed` had been segfaulting on BOLT instrumented binaries. Why I'm not entirely sure. But the segfault only seems to occur in instrumentation mode. These tests are doing low-level things with the interpreter. So I suspect some kind of global mutable state issue or something. I found the exact tests triggering the segfaults and added annotations to skip them. The CPython build system treats the segfault as fatal on 3.13 but not 3.12. This means that on 3.12 we were only running a subset of tests and not collecting BOLT instrumentation nor applying optimizations for all tests after `test_embed`. The removal of the segfault enables us to enable BOLT on 3.13+. Second, LLVM 19.x has a hard error when handling PIC compiled functions containing computed gotos. It appears prior versions of LLVM could silently have buggy behavior in this scenario. We need to skip functions with computed gotos to allow LLVM 19.x to work with BOLT. It makes sense to apply this patch before LLVM 19.x upgrade to prevent bugs with computed gotos. Third, I noticed BOLT was complaining about the lack of `-update-debug-sections` during instrumentation. The 2nd and 3rd issues require common arguments to both BOLT instrumentation and application invocations. The patch fixing both introduces a new configure variable to hold common BOLT arguments. This patch is a good candidate for upstreaming.
1 parent 916bb0d commit 37b0cb6

File tree

3 files changed

+83
-6
lines changed

3 files changed

+83
-6
lines changed

cpython-unix/build-cpython.sh

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -253,6 +253,12 @@ if [ -n "${PYTHON_MEETS_MINIMUM_VERSION_3_11}" ]; then
253253
patch -p1 -i ${ROOT}/patch-pwd-remove-conditional.patch
254254
fi
255255

256+
# Adjust BOLT instrumentation flags to yield better behavior. See inline details
257+
# in patch.
258+
if [ -n "${PYTHON_MEETS_MINIMUM_VERSION_3_12}" ]; then
259+
patch -p1 -i ${ROOT}/patch-configure-bolt-flags.patch
260+
fi
261+
256262
# The optimization make targets are both phony and non-phony. This leads
257263
# to PGO targets getting reevaluated after a build when you use multiple
258264
# make invocations. e.g. `make install` like we do below. Fix that.
@@ -287,6 +293,12 @@ if [ -n "${CROSS_COMPILING}" ]; then
287293
patch -p1 -i ${ROOT}/patch-force-cross-compile.patch
288294
fi
289295

296+
# BOLT instrumented binaries segfault in some test_embed tests for unknown reasons.
297+
# On 3.12 (minimum BOLT version), the segfault causes the test harness to
298+
# abort and BOLT optimization uses the partial test results. On 3.13, the segfault
299+
# is a fatal error.
300+
patch -p1 -i ${ROOT}/patch-test-embed-prevent-segfault.patch
301+
290302
# Most bits look at CFLAGS. But setup.py only looks at CPPFLAGS.
291303
# So we need to set both.
292304
CFLAGS="${EXTRA_TARGET_CFLAGS} -fPIC -I${TOOLS_PATH}/deps/include -I${TOOLS_PATH}/deps/include/ncursesw"
@@ -387,12 +399,7 @@ fi
387399

388400
if [ -n "${CPYTHON_OPTIMIZED}" ]; then
389401
CONFIGURE_FLAGS="${CONFIGURE_FLAGS} --enable-optimizations"
390-
if [[ -n "${PYTHON_MEETS_MINIMUM_VERSION_3_13}" && -n "${BOLT_CAPABLE}" ]]; then
391-
# Due to a SEGFAULT when running `test_embed` with BOLT instrumented binaries, we can't use
392-
# BOLT on Python 3.13+.
393-
# TODO: Find a fix for this or consider skipping these tests specifically
394-
echo "BOLT is disabled on Python 3.13+"
395-
elif [[ -n "${PYTHON_MEETS_MINIMUM_VERSION_3_12}" && -n "${BOLT_CAPABLE}" ]]; then
402+
if [[ -n "${PYTHON_MEETS_MINIMUM_VERSION_3_12}" && -n "${BOLT_CAPABLE}" ]]; then
396403
CONFIGURE_FLAGS="${CONFIGURE_FLAGS} --enable-bolt"
397404
fi
398405
fi
Lines changed: 50 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,50 @@
1+
diff --git a/configure.ac b/configure.ac
2+
index bc8c357e996..eef55d4839a 100644
3+
--- a/configure.ac
4+
+++ b/configure.ac
5+
@@ -2104,6 +2104,27 @@ AS_VAR_IF([enable_shared], [yes], [
6+
BOLT_BINARIES="${BOLT_BINARIES} \$(INSTSONAME)"
7+
])
8+
9+
+AC_ARG_VAR(
10+
+ [BOLT_COMMON_FLAGS],
11+
+ [Common arguments to llvm-bolt when instrumenting and applying]
12+
+)
13+
+
14+
+AC_MSG_CHECKING([BOLT_COMMON_FLAGS])
15+
+if test -z "${BOLT_COMMON_FLAGS}"
16+
+then
17+
+ AS_VAR_SET(
18+
+ [BOLT_COMMON_FLAGS],
19+
+ [m4_normalize("
20+
+ [-update-debug-sections]
21+
+
22+
+ dnl At least LLVM 19.x doesn't support computed gotos in PIC compiled code.
23+
+ dnl Exclude functions containing computed gotos.
24+
+ dnl TODO this may be fixed in LLVM 20.x via https://github.com/llvm/llvm-project/pull/120267.
25+
+ [-skip-funcs=_PyEval_EvalFrameDefault,sre_ucs1_match/1,sre_ucs2_match/1,sre_ucs4_match/1]
26+
+ ")]
27+
+ )
28+
+fi
29+
+
30+
AC_ARG_VAR(
31+
[BOLT_INSTRUMENT_FLAGS],
32+
[Arguments to llvm-bolt when instrumenting binaries]
33+
@@ -2111,7 +2132,7 @@ AC_ARG_VAR(
34+
AC_MSG_CHECKING([BOLT_INSTRUMENT_FLAGS])
35+
if test -z "${BOLT_INSTRUMENT_FLAGS}"
36+
then
37+
- BOLT_INSTRUMENT_FLAGS=
38+
+ BOLT_INSTRUMENT_FLAGS="${BOLT_COMMON_FLAGS}"
39+
fi
40+
AC_MSG_RESULT([$BOLT_INSTRUMENT_FLAGS])
41+
42+
@@ -2125,7 +2146,7 @@ then
43+
AS_VAR_SET(
44+
[BOLT_APPLY_FLAGS],
45+
[m4_normalize("
46+
- -update-debug-sections
47+
+ ${BOLT_COMMON_FLAGS}
48+
-reorder-blocks=ext-tsp
49+
-reorder-functions=hfsort+
50+
-split-functions
Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
diff --git a/Lib/test/test_embed.py b/Lib/test/test_embed.py
2+
index 13713cf37b8..ba23880b15f 100644
3+
--- a/Lib/test/test_embed.py
4+
+++ b/Lib/test/test_embed.py
5+
@@ -1615,6 +1615,7 @@ def test_getpath_abspath_win32(self):
6+
for (_, expected), result in zip(CASES, results):
7+
self.assertEqual(result, expected)
8+
9+
+ @unittest.skip("segfaults on BOLT instrumented binaries")
10+
def test_global_pathconfig(self):
11+
# Test C API functions getting the path configuration:
12+
#
13+
@@ -1866,6 +1867,7 @@ def test_no_memleak(self):
14+
self.assertEqual(blocks, 0, out)
15+
16+
17+
+@unittest.skip("segfaults on BOLT instrumented binaries")
18+
class StdPrinterTests(EmbeddingTestsMixin, unittest.TestCase):
19+
# Test PyStdPrinter_Type which is used by _PySys_SetPreliminaryStderr():
20+
# "Set up a preliminary stderr printer until we have enough

0 commit comments

Comments
 (0)