[SOT] Move Internal Function to External Scope (#71096)

DrRyanHuang · web-flow · commit fb0a9a5c3b22 · 2025-02-12T20:55:19.000+08:00
diff --git a/python/paddle/jit/sot/symbolic/interpreter.py b/python/paddle/jit/sot/symbolic/interpreter.py
@@ -109,16 +109,17 @@ def run_sir(self, name: str, state: dict[str, Symbol]):
         Returns:
             A list of the Symbol of the StatementIR after execution.
         """
+
+        def _set(v, s):
+            state[s.name] = v
+
         SIR = self.get_sir(name)
         for stmt in SIR.statements:
             stmt: Statement
             before_stmt_opnum = opnum_in_program()
             inputs = replace_symbol(stmt.inputs, state)
             outs = getattr(self, stmt.type)(stmt, inputs)
 
-            def _set(v, s):
-                state[s.name] = v
-
             if len(to_sequence(outs)) != len(to_sequence(stmt.outputs)):
                 raise InnerError("Number output mismatch, some error happen.")
 
diff --git a/python/paddle/jit/sot/symbolic/statement_ir.py b/python/paddle/jit/sot/symbolic/statement_ir.py
@@ -98,22 +98,23 @@ def __init__(
         self.type = type
 
     def __str__(self):
-        def to_string(inps):
-            inps = [x.__str__() for x in flatten(inps) if isinstance(x, Symbol)]
-            if len(inps) == 0:
-                return "(Empty)"
-            return ", ".join(inps)
-
         return "{} || {} = {} ({}) ".format(
             self.type + " " * (10 - len(self.type)),
-            to_string(self.outputs),
+            self.to_string(self.outputs),
             self.name,
-            to_string(self.inputs),
+            self.to_string(self.inputs),
         )
 
     def __repr__(self):
         return self.__str__()
 
+    @staticmethod
+    def to_string(inps):
+        inps = [x.__str__() for x in flatten(inps) if isinstance(x, Symbol)]
+        if len(inps) == 0:
+            return "(Empty)"
+        return ", ".join(inps)
+
 
 class CallStatement(Statement):
     def __init__(
@@ -208,7 +209,7 @@ class StatementIR:
     In this way, we can reuse the original `to_static` function to realize the execution of the static graph.
 
     Note:
-        Don't create by yourself, just use the StatementIRCache.get()
+        Don't create by yourself, just use the StatementIRFactory.create()
     """
 
     def __init__(self, name: str):
