def main(): parser = ArgumentParser() parser.add_argument('--cfg', type=str, required=True, help="big daddy config file") args = parser.parse_args() with open(args.cfg) as f: cfg = yaml.safe_load(f) # print(rank, socket.gethostname()) rank = MPI.COMM_WORLD.Get_rank() if rank == 0: m = Master(cfg) m.main() else: w = Worker(cfg) w.main()
def run(workers, foreground, log_file): master = Master(workers, foreground, log_file) exit_code = master.main() exit(exit_code)