Пример #1
0
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()
Пример #2
0
def run(workers, foreground, log_file):
    master = Master(workers, foreground, log_file)
    exit_code = master.main()
    exit(exit_code)