diff --git a/python/paddle/__init__.py b/python/paddle/__init__.py
index 8f5e26bc7c44c085a0f4a35cf61a7a26abc89502..a5a60e58452f886b7cc04a85f3e01c168fe294d3 100755
--- a/python/paddle/__init__.py
+++ b/python/paddle/__init__.py
@@ -535,6 +535,7 @@ __all__ = [  # noqa
     'quantile',
     'nanquantile',
     'no_grad',
+    'enable_grad',
     'set_grad_enabled',
     'is_grad_enabled',
     'mod',