Merge branch 'master' of ssh://github.com/tqchen/rabit

This commit is contained in:
tqchen 2014-12-09 20:56:33 -08:00
commit cc5efb8d81

View File

@ -121,7 +121,7 @@ class SlaveEntry:
return rmset return rmset
class Tracker: class Tracker:
def __init__(self, port = 9000, port_end = 9999): def __init__(self, port = 9091, port_end = 9999):
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
for port in range(port, port_end): for port in range(port, port_end):
try: try:
@ -213,7 +213,7 @@ class Tracker:
assert s.cmd == 'start' or s.cmd == 'recover' assert s.cmd == 'start' or s.cmd == 'recover'
if s.cmd == 'recover': if s.cmd == 'recover':
assert s.rank >= 0 assert s.rank >= 0
print 'Recieve recover signal from %d' % s.rank print 'Recieve %s signal from %d' % (s.cmd, s.rank)
rank = s.decide_rank(job_map) rank = s.decide_rank(job_map)
if rank == -1: if rank == -1:
assert len(todo_nodes) != 0 assert len(todo_nodes) != 0