Skip to content

Commit a557608

Browse files
author
Yan Xu
authored
Merge pull request #13323 from Yancey1989/fix_fluid_benchmark
fix fluid benchmark script
2 parents 68b6192 + b720b3a commit a557608

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

benchmark/fluid/fluid_benchmark.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,8 @@ def dist_transpile(trainer_id, args, train_prog, startup_prog):
9191
program=train_prog,
9292
pservers=pserver_endpoints,
9393
trainers=trainers,
94-
sync_mode=not args.async_mode)
94+
sync_mode=not args.async_mode,
95+
startup_program=startup_prog)
9596
if training_role == "PSERVER":
9697
pserver_program = t.get_pserver_program(current_endpoint)
9798
pserver_startup_program = t.get_startup_program(

0 commit comments

Comments
 (0)