[SOT][3.12] Don't fallback to AST mode in Python 3.12 (#61414)

gouzil · web-flow · commit f4eb8aea9275 · 2024-02-02T14:48:24.000+08:00
diff --git a/python/paddle/jit/api.py b/python/paddle/jit/api.py
@@ -237,9 +237,9 @@ def decorated(python_func):
             flag = ENV_ENABLE_SOT.get()
             full_graph = not flag
 
-        if sys.version_info >= (3, 12) and not full_graph:
+        if sys.version_info >= (3, 13) and not full_graph:
             warnings.warn(
-                "full_graph=False is not supported in Python 3.12+. Set full_graph=True automatically"
+                "full_graph=False is not supported in Python 3.13+. Set full_graph=True automatically"
             )
             full_graph = True
 
diff --git a/test/sot/skip_files_py312 b/test/sot/skip_files_py312
@@ -45,6 +45,7 @@
 ./test_numpy.py
 ./test_numpy_var_if.py
 ./test_output_restoration.py
+./test_segment_linear.py
 ./test_side_effects.py
 ./test_simulate_initialize.py
 ./test_sir_rollback.py
@@ -53,6 +54,7 @@
 ./test_sot_export.py
 ./test_sot_resnet.py
 ./test_sot_resnet50_backward.py
+./test_specialization.py
 ./test_str_format.py
 ./test_tensor_dtype_in_guard.py
 ./test_tensor_slice.py
