From 968b33ec79697d12cace4229aa17e80f25c48d7c Mon Sep 17 00:00:00 2001 From: tqchen Date: Wed, 14 Jan 2015 12:05:00 -0800 Subject: [PATCH] set all tracker thread to deamon --- test/test.mk | 2 +- tracker/rabit_demo.py | 1 + tracker/rabit_tracker.py | 1 + 3 files changed, 3 insertions(+), 1 deletion(-) diff --git a/test/test.mk b/test/test.mk index 05085bdfc..4e0d2b763 100644 --- a/test/test.mk +++ b/test/test.mk @@ -2,7 +2,7 @@ .PHONY: 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 model_recover_10_10k: diff --git a/tracker/rabit_demo.py b/tracker/rabit_demo.py index c7ee04b58..e4592a029 100755 --- a/tracker/rabit_demo.py +++ b/tracker/rabit_demo.py @@ -51,6 +51,7 @@ def mthread_submit(nslave, worker_args): procs = {} for i in range(nslave): procs[i] = Thread(target = exec_cmd, args = (args.command + worker_args, i)) + procs[i].daemon = True procs[i].start() for i in range(nslave): procs[i].join() diff --git a/tracker/rabit_tracker.py b/tracker/rabit_tracker.py index 9823ef426..e260c19c3 100644 --- a/tracker/rabit_tracker.py +++ b/tracker/rabit_tracker.py @@ -257,6 +257,7 @@ class Tracker: def submit(nslave, args, fun_submit, verbose): master = Tracker(verbose = verbose) submit_thread = Thread(target = fun_submit, args = (nslave, args + master.slave_args())) + submit_thread.daemon = True submit_thread.start() master.accept_slaves(nslave) submit_thread.join()