diff --git a/python/setup.py.in b/python/setup.py.in
index a0e665526dc684..61ce98ffda09c3 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',