@@ -96,7 +96,7 @@ def _get_distributed_strategy(self):
96
96
return strategy
97
97
98
98
def _build_trainer_programs (self , compiled_config ):
99
- from paddle .fluid . incubate .fleet .parameter_server .ir import (
99
+ from paddle .incubate .fleet .parameter_server .ir import (
100
100
trainer_pass as worker ,
101
101
)
102
102
@@ -106,7 +106,7 @@ def _build_trainer_programs(self, compiled_config):
106
106
use_ps_gpu = self .user_defined_strategy .a_sync_configs ["use_ps_gpu" ]
107
107
108
108
if not compiled_config .is_geo_mode ():
109
- from paddle .fluid . incubate .fleet .parameter_server .ir .public import (
109
+ from paddle .incubate .fleet .parameter_server .ir .public import (
110
110
_add_lr_decay_table_pass ,
111
111
)
112
112
@@ -150,7 +150,7 @@ def _build_trainer_programs(self, compiled_config):
150
150
compiled_config .set_origin_ps_startup_program (_startup )
151
151
# for heter program
152
152
if self .role_maker ._is_heter_parameter_server_mode :
153
- from paddle .fluid . incubate .fleet .parameter_server .ir import (
153
+ from paddle .incubate .fleet .parameter_server .ir import (
154
154
heter_trainer_pass as heter_worker ,
155
155
)
156
156
@@ -191,13 +191,13 @@ def _build_pserver_programs(self, compiled_config):
191
191
_main = paddle .static .Program ()
192
192
_startup = paddle .static .Program ()
193
193
194
- from paddle .fluid . incubate .fleet .parameter_server .ir import (
194
+ from paddle .incubate .fleet .parameter_server .ir import (
195
195
pserver_pass as server ,
196
196
)
197
197
198
198
if not compiled_config .is_geo_mode ():
199
199
200
- from paddle .fluid . incubate .fleet .parameter_server .ir .public import (
200
+ from paddle .incubate .fleet .parameter_server .ir .public import (
201
201
_get_optimize_ops ,
202
202
)
203
203
@@ -209,7 +209,7 @@ def _build_pserver_programs(self, compiled_config):
209
209
if len (ops ) == 0 :
210
210
return _main , _startup
211
211
212
- from paddle .fluid . incubate .fleet .parameter_server .ir .public import (
212
+ from paddle .incubate .fleet .parameter_server .ir .public import (
213
213
_add_lr_decay_table_pass ,
214
214
)
215
215
@@ -299,9 +299,7 @@ def get_sys_free_mem():
299
299
300
300
free = get_sys_free_mem ()
301
301
302
- from paddle .fluid .incubate .fleet .parameter_server .ir import (
303
- vars_metatools ,
304
- )
302
+ from paddle .incubate .fleet .parameter_server .ir import vars_metatools
305
303
306
304
processed_var_names = set (["@EMPTY@" ])
307
305
param_memory_size = 0
@@ -371,9 +369,7 @@ def minimize_impl(
371
369
372
370
_origin_main_program = loss .block .program
373
371
_origin_startup_program = startup_program
374
- from paddle .fluid .incubate .fleet .parameter_server .ir import (
375
- public as public ,
376
- )
372
+ from paddle .incubate .fleet .parameter_server .ir import public as public
377
373
378
374
compiled_config = public .CompileTimeStrategy (
379
375
_origin_main_program ,
@@ -409,14 +405,14 @@ def minimize_impl(
409
405
}
410
406
else :
411
407
loss .block .program = main_program
412
- fluid .framework .switch_startup_program (startup_program )
408
+ paddle .framework .switch_startup_program (startup_program )
413
409
414
410
elif self .role_maker ._is_server ():
415
411
main_program , startup_program = self ._build_pserver_programs (
416
412
compiled_config
417
413
)
418
414
loss .block .program = main_program
419
- fluid .framework .switch_startup_program (startup_program )
415
+ paddle .framework .switch_startup_program (startup_program )
420
416
return None , None
421
417
422
418
def _disable_strategy (self , dist_strategy ):
0 commit comments