[Dy2St] Add ignore_module API

python/paddle/jit/__init__.py

@@ -17,6 +17,7 @@
 from .api import load
 from .api import to_static
 from .api import not_to_static
+from .api import ignore_module
 from .dy2static.logging_utils import set_code_level, set_verbosity
 
 from . import dy2static
@@ -27,6 +28,7 @@
     'save',
     'load',
     'to_static',
+    'ignore_module',
     'ProgramTranslator',
     'TranslatedLayer',
     'set_code_level',

python/paddle/jit/api.py

@@ -24,7 +24,7 @@
 from collections import OrderedDict
 import inspect
 import threading
-from typing import Any
+from typing import Any, List
 
 import paddle
 from paddle.fluid import core, dygraph
@@ -43,6 +43,7 @@
 from .dy2static.convert_call_func import (
     ConversionOptions,
     CONVERSION_OPTIONS,
+    add_ignore_module,
 )
 from .dy2static.program_translator import (
     ProgramTranslator,
@@ -192,6 +193,34 @@ def copy_decorator_attrs(original_func, decorated_obj):
     return decorated_obj
 
 
+def ignore_module(modules: List[Any]):
+    """
+    Adds modules that ignore transcription.
+    Builtin modules that have been ignored are collections, pdb, copy, inspect, re, numpy, logging, six
+
+    Args:
+        modules (List[Any]): Ignored modules that you want to add
+
+    Examples:
+        .. code-block:: python
+
+            import scipy
+            import astor
+
+            import paddle
+            from paddle.jit import ignore_module
+
+            modules = [
+               scipy,
+               astor
+            ]
+
+            ignore_module(modules)
+
+    """
+    add_ignore_module(modules)
+
+
 def to_static(
     function=None, input_spec=None, build_strategy=None, property=False
 ):

python/paddle/jit/dy2static/convert_call_func.py

@@ -21,6 +21,7 @@
 import pdb
 import re
 import types
+from typing import Any, List
 
 import numpy
 
@@ -100,6 +101,16 @@ def builtin_modules():
 BUILTIN_LIKELY_MODULES = builtin_modules()
 
 
+def add_ignore_module(modules: List[Any]):
+    """
+    Adds modules that ignore transcription
+    """
+    global BUILTIN_LIKELY_MODULES
+    for module in modules:
+        if module not in BUILTIN_LIKELY_MODULES:
+            BUILTIN_LIKELY_MODULES.append(module)
+
+
 def is_unsupported(func):
     """
     Checks whether the func is supported by dygraph to static graph.