From 95983a6218e83538da96fe7a58a01aac22919e32 Mon Sep 17 00:00:00 2001
From: caozhou <48191911+Caozhou1995@users.noreply.github.com>
Date: Thu, 7 Sep 2023 08:26:45 +0800
Subject: [PATCH] add auto tuner module (#56968)

---
 python/setup.py.in | 1 +
 setup.py           | 1 +
 2 files changed, 2 insertions(+)

diff --git a/python/setup.py.in b/python/setup.py.in
index a0e665526dc..61ce98ffda0 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 f2b1df02d8c..962e9d70b00 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',
-- 
GitLab