rename fluid to base (sync with PaddlePaddle/Paddle#56576)

PaddlePaddle · Sep 7, 2023 · 3b97989 · 3b97989
1 parent 32f8634
commit 3b97989
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/sot/infer_meta.py b/sot/infer_meta.py
@@ -1,7 +1,7 @@
 import paddle
 from paddle.amp.auto_cast import amp_state
-from paddle.fluid.unique_name import UniqueNameGenerator
-from paddle.fluid.unique_name import guard as UniqueNameGuard
+from paddle.base.unique_name import UniqueNameGenerator
+from paddle.base.unique_name import guard as UniqueNameGuard
 from paddle.static import Program
 from paddle.utils import flatten, is_sequence
 
@@ -107,7 +107,7 @@ def get_variable(self, meta):
         return self.var_cache[var_feature_name]
 
     def infer_meta(self, func, *args, **kwargs):
-        with paddle.fluid.framework._dygraph_guard(None), UniqueNameGuard(
+        with paddle.base.framework._dygraph_guard(None), UniqueNameGuard(
             self.var_name_generator
         ):
             args, kwargs = convert_meta_to_variable(

diff --git a/sot/opcode_translator/executor/variables/basic.py b/sot/opcode_translator/executor/variables/basic.py
@@ -678,7 +678,7 @@ def main_info(self) -> dict[str, Any]:
 
     @VariableFactory.register_from_value()
     def from_value(value: Any, graph: FunctionGraph, tracker: Tracker):
-        if isinstance(value, paddle.fluid.dygraph.tracer.Tracer):
+        if isinstance(value, paddle.base.dygraph.tracer.Tracer):
             return DygraphTracerVariable(value, graph, tracker)
         return None
 

diff --git a/sot/symbolic/interpreter.py b/sot/symbolic/interpreter.py
@@ -39,7 +39,7 @@ def replace_symbol(
 def _append_opstack_between(start, end, stack):
     # NOTE(xiongkun): we don't sync for speed. careful!!
     # [start, end)
-    from paddle.fluid import core
+    from paddle.framework import core
 
     op_maker = core.op_proto_and_checker_maker
     callstack_attr_name = op_maker.kOpCreationCallstackAttrName()

diff --git a/sot/utils/utils.py b/sot/utils/utils.py
@@ -255,7 +255,7 @@ class GraphLogger:
     graph_num: int
     op_num: int
     graphs: list[Program]
-    ops: list[paddle.fluid.framework.Operator]
+    ops: list[paddle.base.framework.Operator]
 
     def __init__(self):
         self.clear()