[SOT] Add SOT_ENABLE_0_SIZE_FALLBACK environment variable (#73399)

0x3878f · SigureMo · web-flow · commit 47b355970133 · 2025-06-18T20:04:41.000+08:00
---------

Co-authored-by: Nyakku Shigure &lt;sigure.qaq@gmail.com&gt;
diff --git a/python/paddle/jit/sot/opcode_translator/executor/variables/basic.py b/python/paddle/jit/sot/opcode_translator/executor/variables/basic.py
@@ -76,6 +76,7 @@
 )
 from ....utils import (
     ENV_SOT_ALLOW_DYNAMIC_SHAPE,
+    ENV_SOT_ENABLE_0_SIZE_FALLBACK,
     BreakGraphError,
     BuiltinFunctionBreak,
     ConditionalFallbackError,
@@ -1446,6 +1447,7 @@ def from_value(value: Any, graph: FunctionGraph, tracker: Tracker):
         )
         if (
             should_create_sym is None
+            and ENV_SOT_ENABLE_0_SIZE_FALLBACK.get()
             and SymbolicVariable.find_tensor_shape_source(tracker) is not None
         ):
             graph.add_global_guarded_variable(
diff --git a/python/paddle/jit/sot/utils/__init__.py b/python/paddle/jit/sot/utils/__init__.py
@@ -18,6 +18,7 @@
     ENV_SOT_ALLOW_DYNAMIC_SHAPE,
     ENV_SOT_CE_DEBUG_MODE,
     ENV_SOT_COLLECT_INFO,
+    ENV_SOT_ENABLE_0_SIZE_FALLBACK,
     ENV_SOT_ENABLE_FASTER_GUARD,
     ENV_SOT_ENABLE_GUARD_TREE,
     ENV_SOT_ENABLE_STRICT_GUARD_CHECK,
diff --git a/python/paddle/jit/sot/utils/envs.py b/python/paddle/jit/sot/utils/envs.py
@@ -153,6 +153,9 @@ def parse_parameterized_key(input_str: str) -> dict[str, list[str]]:
 ENV_SOT_UNSAFE_CACHE_FASTPATH = BooleanEnvironmentVariable(
     "SOT_UNSAFE_CACHE_FASTPATH", False
 )
+ENV_SOT_ENABLE_0_SIZE_FALLBACK = BooleanEnvironmentVariable(
+    "SOT_ENABLE_0_SIZE_FALLBACK", True
+)
 
 
 def update_ce_flags():

Original file line number	Diff line number	Diff line change
`@@ -153,6 +153,9 @@ def parse_parameterized_key(input_str: str) -> dict[str, list[str]]:`
`153`	`153`	`ENV_SOT_UNSAFE_CACHE_FASTPATH = BooleanEnvironmentVariable(`
`154`	`154`	`"SOT_UNSAFE_CACHE_FASTPATH", False`
`155`	`155`	`)`
	`156`	`+ENV_SOT_ENABLE_0_SIZE_FALLBACK = BooleanEnvironmentVariable(`
	`157`	`+ "SOT_ENABLE_0_SIZE_FALLBACK", True`
	`158`	`+)`
`156`	`159`
`157`	`160`
`158`	`161`	`def update_ce_flags():`