Skip to content

Commit 8f83d20

Browse files
authored
[Dy2St] Cleanup unused methods in ProgramTranslator (#58302)
1 parent 0ad330e commit 8f83d20

File tree

2 files changed

+28
-333
lines changed

2 files changed

+28
-333
lines changed

python/paddle/jit/api.py

Lines changed: 28 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -15,18 +15,17 @@
1515

1616
# Temporary disable isort to avoid circular import
1717
# This can be removed after the circular import is resolved
18-
# isort: skip_file
1918
from __future__ import annotations
2019

20+
import inspect
2121
import os
2222
import pickle
23-
import warnings
2423
import sys
25-
from collections import OrderedDict
26-
import inspect
2724
import threading
28-
from typing import Any
2925
import types
26+
import warnings
27+
from collections import OrderedDict
28+
from typing import Any
3029

3130
import paddle
3231
from paddle.base import core, dygraph
@@ -40,43 +39,40 @@
4039
program_desc_tracing_guard,
4140
switch_to_static_graph,
4241
)
43-
from .dy2static import logging_utils
44-
from .dy2static.convert_call_func import (
45-
ConversionOptions,
46-
add_ignore_module,
47-
)
48-
from .dy2static.program_translator import (
49-
ProgramTranslator,
50-
StaticFunction,
51-
ASTStaticFunction,
52-
SymbolicStaticFunction,
53-
unwrap_decorators,
54-
)
55-
from paddle.jit.translated_layer import (
56-
TranslatedLayer,
57-
INFER_MODEL_SUFFIX,
58-
INFER_PARAMS_SUFFIX,
59-
INFER_PARAMS_INFO_SUFFIX,
60-
INFER_PROPERTY_SUFFIX,
61-
)
62-
from paddle.nn import Layer
6342
from paddle.base.executor import Executor, scope_guard
6443
from paddle.base.framework import (
6544
Block,
45+
EagerParamBase,
46+
Parameter,
6647
Program,
6748
Variable,
68-
Parameter,
69-
EagerParamBase,
70-
)
71-
from paddle.base.framework import (
7249
_current_expected_place,
7350
_dygraph_guard,
7451
_dygraph_tracer,
52+
dygraph_only,
7553
)
76-
from paddle.base.framework import dygraph_only
7754
from paddle.base.wrapped_decorator import wrap_decorator
78-
from paddle.static.io import save_inference_model
7955
from paddle.framework import in_dynamic_mode
56+
from paddle.nn import Layer
57+
from paddle.static.io import save_inference_model
58+
59+
from .dy2static import logging_utils
60+
from .dy2static.convert_call_func import ConversionOptions, add_ignore_module
61+
from .dy2static.program_translator import (
62+
ASTStaticFunction,
63+
ProgramTranslator,
64+
StaticFunction,
65+
SymbolicStaticFunction,
66+
convert_to_static,
67+
unwrap_decorators,
68+
)
69+
from .translated_layer import (
70+
INFER_MODEL_SUFFIX,
71+
INFER_PARAMS_INFO_SUFFIX,
72+
INFER_PARAMS_SUFFIX,
73+
INFER_PROPERTY_SUFFIX,
74+
TranslatedLayer,
75+
)
8076

8177

8278
def create_program_from_desc(program_desc):
@@ -166,7 +162,7 @@ def __impl__(*args, **kwargs):
166162
"We will just return dygraph output."
167163
)
168164
return dygraph_func(*args, **kwargs)
169-
static_func = program_translator.get_func(dygraph_func)
165+
static_func = convert_to_static(dygraph_func)
170166
return static_func(*args, **kwargs)
171167

172168
return __impl__

0 commit comments

Comments
 (0)