in src/sagemaker_xgboost_container/dmlc_patch/tracker.py [0:0]
def __init__(self, sock, s_addr):
slave = ExSocket(sock)
self.sock = slave
self.host = get_some_ip(s_addr[0])
magic = slave.recvint()
assert magic == kMagic, "invalid magic number=%d from %s" % (magic, self.host)
slave.sendint(kMagic)
self.rank = slave.recvint()
logger.debug("{}:{} Rank {}".format(self.sock, s_addr, self.rank))
self.world_size = slave.recvint()
logger.debug("{}:{} world_size {}".format(self.sock, s_addr, self.world_size))
self.jobid = slave.recvstr()
logger.debug("{}:{} jobid {}".format(self.sock, s_addr, self.jobid))
self.cmd = slave.recvstr()
logger.debug("{}:{} cmd {}".format(self.sock, s_addr, self.cmd))
self.wait_accept = 0
self.port = None