Permalink
Browse files

add hostip option is dmlc_mpi

  • Loading branch information...
1 parent f25ed36 commit f2ab107e2d72123d653633e60be2202bd59a5432 @mli mli committed Mar 22, 2016
Showing with 3 additions and 0 deletions.
  1. +3 −0 tracker/dmlc_mpi.py
View
@@ -23,6 +23,8 @@
help = 'the hostfile of mpi server')
parser.add_argument('command', nargs='+',
help = 'command for dmlc program')
+parser.add_argument('--host-ip', type=str,
+ help = 'the scheduler ip', default='ip')
args, unknown = parser.parse_known_args()
#
# submission script using MPI
@@ -85,4 +87,5 @@ def run(prog):
tracker.config_logger(args)
tracker.submit(args.nworker, args.server_nodes, fun_submit = mpi_submit,
+ hostIP=args.host_ip,
pscmd=(' '.join(args.command) + ' ' + ' '.join(unknown)))

0 comments on commit f2ab107

Please sign in to comment.