diff --git a/python/setup.py.in b/python/setup.py.in
index a0e665526dc684ec02e5da19de43132ed2a090cf..61ce98ffda09c3168e32e1669517e21fcbb6f2b8 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 f2b1df02d8c1e4844716a542fc075b871fdb3e55..962e9d70b002a09387500bdbcbf60f18c858c29c 100644
--- a/setup.py
+++ b/setup.py
@@ -1377,6 +1377,7 @@ def get_setup_parameters():
         'paddle.distributed.utils',
         'paddle.distributed.sharding',
         'paddle.distributed.fleet',
+        'paddle.distributed.auto_tuner',
         'paddle.distributed.launch',
         'paddle.distributed.launch.context',
         'paddle.distributed.launch.controllers',