[Dy2St] Cleanup IrParser based subgraph exporter (#70569)

SigureMo · web-flow · commit c5a5c3f8da64 · 2025-01-02T13:11:06.000+08:00
diff --git a/python/paddle/jit/dy2static/export_subgraph.py b/python/paddle/jit/dy2static/export_subgraph.py
diff --git a/python/paddle/jit/dy2static/partial_program.py b/python/paddle/jit/dy2static/partial_program.py
@@ -27,7 +27,6 @@
 from paddle.optimizer.lr import LRScheduler
 
 from . import logging_utils
-from .export_subgraph import SubGraphRole, pir_exporter
 from .utils import (
     RETURN_NO_VALUE_MAGIC_NUM,
     backend_guard,
@@ -214,7 +213,6 @@ def __init__(
         self._out_var_descs = [
             self._outputs[var_id].desc for var_id in self._outputs.var_ids
         ]
-        self._debug_name = None
 
     def __call__(self, inputs):
         """
@@ -517,14 +515,11 @@ def infer_program(self):
             infer_program = self._infer_pure_fp16_program
         else:
             infer_program = self._infer_program
-        # NOTE(Aurelius84): Export forward_program for SubGraphChecker,
-        # see export_subgraph for detail.
-        pir_exporter(self, infer_program, SubGraphRole.Infer)
         return infer_program
 
     @property
     def forward_program(self):
-        forward_program, role = None, None
+        forward_program = None
         if self.training:
             if _in_amp_guard():
                 progs = self._train_amp_forward_backward_program
@@ -854,22 +849,6 @@ def _get_forward_backward_program_form(
 
         self._apply_inplace_pass(forward_built_program, backward_built_program)
 
-        # NOTE(Aurelius84): Export forward/backward program for SubGraphChecker,
-        # see export_subgraph for detail.
-        pir_exporter(
-            self,
-            forward_built_program,
-            SubGraphRole.Forward,
-            set(),
-            set(forward_skip_vars),
-        )
-        pir_exporter(
-            self,
-            backward_built_program,
-            SubGraphRole.Backward,
-            set(forward_skip_vars),
-            set(backward_skip_vars),
-        )
         return [forward_built_program, backward_built_program]
 
     def _apply_inplace_pass(self, forward_program, backward_program):
diff --git a/python/paddle/jit/dy2static/pir_partial_program.py b/python/paddle/jit/dy2static/pir_partial_program.py
@@ -619,7 +619,6 @@ def __init__(
         self._hookers = []
         self._backend = kwargs.get('backend', None)
         self._grad_var_names = {}
-        self._debug_name = None
 
     def __call__(self, inputs):
         """
diff --git a/python/paddle/jit/dy2static/program_translator.py b/python/paddle/jit/dy2static/program_translator.py
@@ -436,16 +436,6 @@ def __init__(self, function, input_spec=None, **kwargs):
         self._property = kwargs.get("property", False)
         # Note: Record the patched method name for rollback.
         self._patched_name = None
-        self._get_debug_name()
-
-    def _get_debug_name(self) -> str:
-        try:
-            if self.class_instance:
-                self._debug_name = self.class_instance.__class__.__name__
-            else:
-                self._debug_name = self._dygraph_function.__name__
-        except Exception:
-            self._debug_name = "static_function"
 
     @property
     def is_property(self) -> bool:
@@ -947,7 +937,6 @@ def get_concrete_program(
             concrete_program, partial_program_layer = self._program_cache[
                 cache_key
             ]
-        partial_program_layer._debug_name = self._debug_name
         return concrete_program, partial_program_layer
 
     def get_concrete_program_with_cache_key(
