[SOT][DyShape] add ENV_SOT_ALLOW_DYNAMIC_SHAPE (#64278)

zrr1999 · web-flow · commit a9f14a3b4295 · 2024-05-15T08:55:25.000+08:00
diff --git a/python/paddle/jit/sot/utils/envs.py b/python/paddle/jit/sot/utils/envs.py
@@ -33,6 +33,9 @@
     "SOT_WITH_CONTROL_FLOW", True
 )
 ENV_SOT_EXPORT = StringEnvironmentVariable("SOT_EXPORT", "")
+ENV_SOT_ALLOW_DYNAMIC_SHAPE = BooleanEnvironmentVariable(
+    "SOT_ALLOW_DYNAMIC_SHAPE", False
+)
 
 
 @contextmanager

Original file line number	Diff line number	Diff line change
`@@ -33,6 +33,9 @@`
`33`	`33`	`"SOT_WITH_CONTROL_FLOW", True`
`34`	`34`	`)`
`35`	`35`	`ENV_SOT_EXPORT = StringEnvironmentVariable("SOT_EXPORT", "")`
	`36`	`+ENV_SOT_ALLOW_DYNAMIC_SHAPE = BooleanEnvironmentVariable(`
	`37`	`+ "SOT_ALLOW_DYNAMIC_SHAPE", False`
	`38`	`+)`
`36`	`39`
`37`	`40`
`38`	`41`	`@contextmanager`