test.support patches

RustPython · arihant2math · Apr 27, 2025 · Apr 27, 2025 · Apr 27, 2025 · Apr 27, 2025
commit c03ff666826573dfbf7ff4defc90c5c2be704f1d
diff --git a/Lib/test/support/__init__.py b/Lib/test/support/__init__.py
@@ -837,6 +837,30 @@ def python_is_optimized():
             final_opt = opt
     return final_opt not in ('', '-O0', '-Og')
 
+def check_cflags_pgo():
+    # Check if Python was built with ./configure --enable-optimizations:
+    # with Profile Guided Optimization (PGO).
+    cflags_nodist = sysconfig.get_config_var('PY_CFLAGS_NODIST') or ''
+    pgo_options = [
+        # GCC
+        '-fprofile-use',
+        # clang: -fprofile-instr-use=code.profclangd
+        '-fprofile-instr-use',
+        # ICC
+        "-prof-use",
+    ]
+    PGO_PROF_USE_FLAG = sysconfig.get_config_var('PGO_PROF_USE_FLAG')
+    if PGO_PROF_USE_FLAG:
+        pgo_options.append(PGO_PROF_USE_FLAG)
+    return any(option in cflags_nodist for option in pgo_options)
+
+def check_bolt_optimized():
+    # Always return false, if the platform is WASI,
+    # because BOLT optimization does not support WASM binary.
+    if is_wasi:
+        return False
+    config_args = sysconfig.get_config_var('CONFIG_ARGS') or ''
+    return '--enable-bolt' in config_args
 
 _header = 'nP'
 _align = '0n'