2023-04-04 11:47:36 +00:00

33 lines
652 B
C++

#include <chrono>
#include <thread>
#include <fstream>
#include <mpi.h>
#include "light.hpp"
#include "utils/cmdline.h"
#include "paras.hpp"
#include "distributed/leader.hpp"
#include "distributed/worker.hpp"
INITIALIZE_EASYLOGGINGPP
int main(int argc, char **argv) {
int num_procs, rank;
MPI_Init(&argc, &argv);
MPI_Comm_size(MPI_COMM_WORLD, &num_procs);
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
light* S = new light();
S->arg_parse(argc, argv);
// leader
if(rank == 0) leader_main(S, num_procs, rank);
else worker_main(S, num_procs, rank);
MPI_Barrier(MPI_COMM_WORLD);
MPI_Finalize();
return 0;
}