[Fluid Clean] Migrate program_translate.py/jit.py into paddle.jit dir (#48240)

* [Fluid Clean] Migrate program_translate.py/jit.py into paddle.jit dir

Author: Aurelius84

.gitignore

@@ -75,6 +75,7 @@ paddle/fluid/operators/generated_op.cc
 paddle/fluid/operators/generated_sparse_op.cc
 paddle/phi/ops/compat/generated_sig.cc
 paddle/phi/ops/compat/generated_sparse_sig.cc
+paddle/phi/api/yaml/parsed_apis/
 paddle/fluid/operators/generator/parsed_ops/
 paddle/fluid/pybind/tmp_eager_op_function_impl.h
 paddle/fluid/pybind/eager_op_function_impl.h

python/paddle/distributed/auto_parallel/helper.py

@@ -20,7 +20,7 @@
 from paddle.fluid.framework import Parameter
 from paddle.fluid.framework import program_guard
 from paddle.fluid.executor import global_scope
-from paddle.fluid.dygraph.dygraph_to_static.program_translator import (
+from paddle.jit.dy2static.program_translator import (
     StaticFunction,
 )
 

python/paddle/fluid/compiler.py

@@ -691,16 +691,12 @@ def patch_program_cache(ipu_strategy):
         Returns:
             None
         """
-        from ..fluid.dygraph.dygraph_to_static.program_translator import (
-            ProgramCache,
-        )
-        from ..fluid.dygraph.dygraph_to_static.program_translator import (
+        from paddle.jit.dy2static.program_translator import (
             CacheKey,
-        )
-        from ..fluid.dygraph.dygraph_to_static import logging_utils
-        from ..fluid.dygraph.dygraph_to_static.program_translator import (
+            ProgramCache,
             MAX_TRACED_PROGRAM_COUNT,
         )
+        from ..fluid.dygraph.dygraph_to_static import logging_utils
         from ..fluid.dygraph.dygraph_to_static.partial_program import (
             partial_program_from,
         )

python/paddle/fluid/dygraph/__init__.py

@@ -36,8 +36,6 @@
 from . import learning_rate_scheduler
 from .learning_rate_scheduler import *
 
-from . import jit
-from .jit import *
 
 from . import io
 from .io import *
@@ -46,7 +44,6 @@
 from .static_runner import StaticModelRunner
 
 from . import dygraph_to_static
-from .dygraph_to_static import ProgramTranslator
 
 from . import rnn
 from .rnn import *
@@ -66,8 +63,6 @@
 __all__ += parallel.__all__
 __all__ += checkpoint.__all__
 __all__ += learning_rate_scheduler.__all__
-__all__ += jit.__all__
 __all__ += io.__all__
 __all__ += rnn.__all__
-__all__ += ['ProgramTranslator']
 __all__ += amp.__all__

python/paddle/fluid/dygraph/checkpoint.py

@@ -31,7 +31,7 @@
 import warnings
 from .. import core
 from .base import guard
-from paddle.fluid.dygraph.jit import _SaveLoadConfig
+from paddle.jit.api import _SaveLoadConfig
 from paddle.fluid.dygraph.io import (
     _construct_program_holders,
     _construct_params_and_buffers,

python/paddle/fluid/dygraph/dygraph_to_static/__init__.py

@@ -12,9 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from . import ast_transformer
-from .ast_transformer import *
-
 from . import static_analysis
 from .static_analysis import *
 
@@ -24,9 +21,6 @@
 from . import variable_trans_func
 from .variable_trans_func import *
 
-from . import program_translator
-from .program_translator import *
-
 from . import convert_call_func
 from .convert_call_func import *
 
@@ -36,10 +30,8 @@
 from .logging_utils import *
 
 __all__ = []
-__all__ += ast_transformer.__all__
 __all__ += loop_transformer.__all__
 __all__ += static_analysis.__all__
 __all__ += variable_trans_func.__all__
-__all__ += program_translator.__all__
 __all__ += convert_call_func.__all__
 __all__ += logging_utils.__all__

python/paddle/fluid/dygraph/dygraph_to_static/convert_call_func.py

@@ -36,15 +36,7 @@
 from paddle.fluid.dygraph.dygraph_to_static.logging_utils import (
     TranslatorLogger,
 )
-from paddle.fluid.dygraph.dygraph_to_static.program_translator import (
-    StaticFunction,
-)
-from paddle.fluid.dygraph.dygraph_to_static.program_translator import (
-    convert_to_static,
-)
-from paddle.fluid.dygraph.dygraph_to_static.program_translator import (
-    unwrap_decorators,
-)
+
 from paddle.fluid.dygraph.dygraph_to_static.utils import is_paddle_func, unwrap
 from paddle.fluid.dygraph.layers import Layer
 
@@ -185,6 +177,13 @@ def dyfunc(x):
             #  [1. 1. 1.]]
 
     """
+    # NOTE(Aurelius84): Fix it after all files migrating into jit.
+    from paddle.jit.dy2static.program_translator import (
+        convert_to_static,
+        unwrap_decorators,
+        StaticFunction,
+    )
+
     translator_logger.log(
         1, "Convert callable object: convert {}.".format(func)
     )

python/paddle/fluid/dygraph/static_runner.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from paddle.fluid.dygraph.jit import _SaveLoadConfig
+from paddle.jit.api import _SaveLoadConfig
 from paddle.fluid.dygraph.io import TranslatedLayer
 
 

python/paddle/fluid/tests/unittests/dygraph_to_static/bert_dygraph_model.py

@@ -15,7 +15,7 @@
 import paddle
 import paddle.fluid as fluid
 from paddle.fluid.dygraph import Embedding, Layer, Linear
-from paddle.fluid.dygraph.jit import declarative
+from paddle.jit.api import declarative
 
 from transformer_dygraph_model import MultiHeadAttention, PrePostProcessLayer
 

python/paddle/fluid/tests/unittests/dygraph_to_static/seq2seq_dygraph_model.py

@@ -20,7 +20,7 @@
 from paddle.fluid import layers
 from paddle.fluid.dygraph import Layer
 from paddle.fluid.dygraph.base import to_variable
-from paddle.fluid.dygraph.jit import declarative
+from paddle.jit.api import declarative
 from paddle.fluid.dygraph.nn import Embedding
 from seq2seq_utils import Seq2SeqModelHyperParams as args