Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

set all tracker thread to deamon

Browse Source
This commit is contained in:
tqchen 2015-01-14 12:05:00 -08:00
parent 87c7817124
commit 968b33ec79
3 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/test.mk
View File

@ -2,7 +2,7 @@
.PHONY: .PHONY:
test: test:
../tracker/rabit_mpi.py -v 1 -n 10 bash keepalive.sh test_model_recover 1 mock=0,0,1,0 mock=1,1,1,0 mock=1,1,1,1 mock=0,1,1,0 mock=4,1,1,0 mock=8,1,2,0 ../tracker/rabit_demo.py -v 1 -n 10 test_model_recover 1 mock=0,0,1,0 mock=1,1,1,0 mock=1,1,1,1 mock=0,1,1,0 mock=4,1,1,0 mock=8,1,2,0
# this experiment test recovery with actually process exit, use keepalive to keep program alive # this experiment test recovery with actually process exit, use keepalive to keep program alive
model_recover_10_10k: model_recover_10_10k:

1
tracker/rabit_demo.py
View File

@ -51,6 +51,7 @@ def mthread_submit(nslave, worker_args):
procs = {} procs = {}
for i in range(nslave): for i in range(nslave):
procs[i] = Thread(target = exec_cmd, args = (args.command + worker_args, i)) procs[i] = Thread(target = exec_cmd, args = (args.command + worker_args, i))
procs[i].daemon = True
procs[i].start() procs[i].start()
for i in range(nslave): for i in range(nslave):
procs[i].join() procs[i].join()

1
tracker/rabit_tracker.py
View File

@ -257,6 +257,7 @@ class Tracker:
def submit(nslave, args, fun_submit, verbose): def submit(nslave, args, fun_submit, verbose):
master = Tracker(verbose = verbose) master = Tracker(verbose = verbose)
submit_thread = Thread(target = fun_submit, args = (nslave, args + master.slave_args())) submit_thread = Thread(target = fun_submit, args = (nslave, args + master.slave_args()))
submit_thread.daemon = True
submit_thread.start() submit_thread.start()
master.accept_slaves(nslave) master.accept_slaves(nslave)
submit_thread.join() submit_thread.join()