cleanup env cost model

cattidea · Feb 26, 2025 · 22d06e3 · 22d06e3
1 parent 4af8662
commit 22d06e3
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 9 deletions.
diff --git a/python/paddle/jit/sot/utils/__init__.py b/python/paddle/jit/sot/utils/__init__.py
@@ -14,7 +14,6 @@
 
 from .call_ast_utils import get_static_function, try_ast_func  # noqa: F401
 from .envs import (  # noqa: F401
-    ENV_COST_MODEL,
     ENV_MIN_GRAPH_SIZE,
     ENV_SOT_ALLOW_DYNAMIC_SHAPE,
     ENV_SOT_ENABLE_FASTER_GUARD,
@@ -25,7 +24,6 @@
     ENV_SOT_WITH_CONTROL_FLOW,
     ENV_STRICT_MODE,
     allow_dynamic_shape_guard,
-    cost_model_guard,
     export_guard,
     faster_guard_guard,
     guard_tree_guard,

diff --git a/python/paddle/jit/sot/utils/envs.py b/python/paddle/jit/sot/utils/envs.py
@@ -25,7 +25,6 @@
     StringListEnvironmentVariable,
 )
 
-ENV_COST_MODEL = BooleanEnvironmentVariable("COST_MODEL", False)
 ENV_MIN_GRAPH_SIZE = IntegerEnvironmentVariable("MIN_GRAPH_SIZE", 10)
 ENV_SOT_LOG_LEVEL = IntegerEnvironmentVariable("SOT_LOG_LEVEL", 0)
 ENV_STRICT_MODE = BooleanEnvironmentVariable("STRICT_MODE", False)
@@ -59,12 +58,6 @@
 )
 
 
-@contextmanager
-def cost_model_guard(value: bool):
-    with EnvironmentVariableGuard(ENV_COST_MODEL, value):
-        yield
-
-
 @contextmanager
 def strict_mode_guard(value: bool):
     with EnvironmentVariableGuard(ENV_STRICT_MODE, value):