loadbal/main.c at master · cocolikestocode/loadbal · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
#include <mpi.h>
#include <loadbal.h>
#include <assist.h>

int main (int argc, char *argv[])
{
	int rank, np, *nbrs, j, *jobsmap;
	uint alljobsnum, actjobsnum;
	uint nbredgenum, nbrsnum;
	job_t *alljobs;
	uint *activejobs;
	MPI_Request *sharereqs, *wloadreqs;
	int errnum;

	errnum = MPI_Init(&argc, &argv);
	if (errnum != MPI_SUCCESS)
		PRERROR("main: cannot initialize: ", errnum);
	errnum = MPI_Comm_size(MPI_COMM_WORLD, &np);
	if (errnum != MPI_SUCCESS)
		PRERROR("main: cannot get size: ", errnum);
	errnum = MPI_Comm_rank(MPI_COMM_WORLD, &rank);
	if (errnum != MPI_SUCCESS)
		PRERROR("main: cannot get process rank: ", errnum);

	assist_init();
	alljobs = form_jobs(np, &alljobsnum);
	jobsmap = calloc(alljobsnum, sizeof(int));
	activejobs = distr_jobs1(rank, np, alljobs, alljobsnum, &actjobsnum,
			jobsmap);
	/*
	printf("%d(%d): ", rank, actjobsnum);
	for (j = 0; j < actjobsnum; j++)
		printf("%d, ", activejobs[j]);
	printf("\n");
	fflush(stdout);
	*/

	alloc_memory(alljobs, activejobs, actjobsnum);
	set_init_cond(alljobs, activejobs, actjobsnum);
	//draw(rank, np, alljobs, activejobs, actjobsnum);
	nbredgenum = count_nbredges(alljobs, activejobs, actjobsnum, rank);
	sharereqs = prep_shrreqs(nbredgenum);
	nbrs = get_nbrs(rank, np, alljobs, activejobs, actjobsnum, &nbrsnum);
	wloadreqs = prep_wldreqs(nbrsnum);

	double mes;
	for (j = 1; j <= 30000; j++) {
		make_timestep(alljobs, activejobs, actjobsnum, sharereqs,
				nbredgenum, rank);
		if ((j%50) == 0) {
			mes = mes_disb(rank, np, alljobs, activejobs, actjobsnum);
			if (rank == 0) {
				printf("mes = %f\n", mes);
				fflush(stdout);
			}
			if (mes > CRIT_MES) {
				rebalance(rank, np, alljobs, alljobsnum, &activejobs,
					&actjobsnum, nbrs, nbrsnum, jobsmap, wloadreqs);
				renew_resources(rank, np, alljobs, activejobs, actjobsnum,
					&sharereqs, &wloadreqs, &nbredgenum,
					&nbrs, &nbrsnum);
			}
			else
				nullify_wloads(alljobs, activejobs, actjobsnum);
		}
	}

	free_resources(alljobs, activejobs, actjobsnum, sharereqs, wloadreqs,
			nbrs, jobsmap, rank);
	errnum =  MPI_Finalize();
	if (errnum != MPI_SUCCESS)
		PRERROR("main: cannot finalize: ", errnum);
	return EXIT_SUCCESS;
}