add auto tuner module

PaddlePaddle · Sep 6, 2023 · 837d28f · 837d28f
1 parent 1a15a35
commit 837d28f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -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',