diff --git a/python/setup.py.in b/python/setup.py.in
index a0e665526dc68..61ce98ffda09c 100644
--- a/python/setup.py.in
+++ b/python/setup.py.in
@@ -380,6 +380,7 @@ packages=['paddle',
           'paddle.distributed.sharding',
           'paddle.distributed.fleet',
           'paddle.distributed.launch',
+          'paddle.distributed.auto_tuner',
           'paddle.distributed.launch.context',
           'paddle.distributed.launch.controllers',
           'paddle.distributed.launch.job',
diff --git a/setup.py b/setup.py
index f2b1df02d8c1e..9545943cf3d64 100644
--- a/setup.py
+++ b/setup.py
@@ -1378,6 +1378,7 @@ def get_setup_parameters():
         'paddle.distributed.sharding',
         'paddle.distributed.fleet',
         'paddle.distributed.launch',
+        'paddle.distributed.auto_tuner',
         'paddle.distributed.launch.context',
         'paddle.distributed.launch.controllers',
         'paddle.distributed.launch.job',