gh-124613: Deactivate the JIT during perf tests

pablogsal · pablogsal · commit 036c09bb911e · 2024-10-01T01:01:05.000+01:00
diff --git a/Lib/test/test_perf_profiler.py b/Lib/test/test_perf_profiler.py
@@ -23,15 +23,6 @@
     raise unittest.SkipTest("test crash randomly on ASAN/MSAN/UBSAN build")
 
 
-def is_jit_build():
-    cflags = (sysconfig.get_config_var("PY_CORE_CFLAGS") or '')
-    return "_Py_JIT" in cflags
-
-
-if is_jit_build():
-    raise unittest.SkipTest("Perf support is not available in JIT builds")
-
-
 def supports_trampoline_profiling():
     perf_trampoline = sysconfig.get_config_var("PY_HAVE_PERF_TRAMPOLINE")
     if not perf_trampoline:
@@ -58,12 +49,15 @@ def tearDown(self) -> None:
 
     def test_trampoline_works(self):
         code = """if 1:
+                from test import support
+
                 def foo():
                     pass
 
                 def bar():
                     foo()
 
+                @support.without_optimizer
                 def baz():
                     bar()
 
@@ -110,6 +104,7 @@ def baz():
     def test_trampoline_works_with_forks(self):
         code = """if 1:
                 import os, sys
+                from test import support
 
                 def foo_fork():
                     pass
@@ -132,6 +127,7 @@ def foo():
                 def bar():
                     foo()
 
+                @support.without_optimizer
                 def baz():
                     bar()
 
@@ -168,6 +164,8 @@ def baz():
     def test_sys_api(self):
         code = """if 1:
                 import sys
+                from test import support
+
                 def foo():
                     pass
 
@@ -180,6 +178,7 @@ def bar():
                     sys.activate_stack_trampoline("perf")
                     spam()
 
+                @support.without_optimizer
                 def baz():
                     bar()
 
@@ -344,6 +343,8 @@ def run_perf(self, script_dir, perf_mode, script):
     def test_python_calls_appear_in_the_stack_if_perf_activated(self):
         with temp_dir() as script_dir:
             code = """if 1:
+                from test import support
+
                 def foo(n):
                     x = 0
                     for i in range(n):
@@ -352,6 +353,7 @@ def foo(n):
                 def bar(n):
                     foo(n)
 
+                @support.without_optimizer
                 def baz(n):
                     bar(n)
 
@@ -368,6 +370,8 @@ def baz(n):
     def test_python_calls_do_not_appear_in_the_stack_if_perf_deactivated(self):
         with temp_dir() as script_dir:
             code = """if 1:
+                from test import support
+
                 def foo(n):
                     x = 0
                     for i in range(n):
@@ -376,6 +380,7 @@ def foo(n):
                 def bar(n):
                     foo(n)
 
+                @support.without_optimizer
                 def baz(n):
                     bar(n)
 
@@ -428,12 +433,14 @@ def test_pre_fork_compile(self):
                 def foo_fork():
                     pass
 
+                @support.without_optimizer
                 def bar_fork():
                     foo_fork()
 
                 def foo():
                     import time; time.sleep(1)
 
+                @support.without_optimizer
                 def bar():
                     foo()
 
@@ -442,7 +449,6 @@ def compile_trampolines_for_all_functions():
                     for _, obj in globals().items():
                         if callable(obj) and hasattr(obj, '__code__'):
                             compile_perf_trampoline_entry(obj.__code__)
-
                 if __name__ == "__main__":
                     compile_trampolines_for_all_functions()
                     pid = os.fork()
