1 | #include <mpi.h> |
---|
2 | #include <stdio.h> |
---|
3 | #include <parmetis.h> |
---|
4 | |
---|
5 | void dynamico_partition_graph(int mpi_rank, int mpi_size, |
---|
6 | idx_t vtxdist[], idx_t xadj_loc[], idx_t adjncy_loc[], |
---|
7 | idx_t nparts, idx_t part[]) |
---|
8 | { |
---|
9 | MPI_Comm comm = MPI_COMM_WORLD; /* avoid passing a communicator as argument due to Fortran/C issues */ |
---|
10 | idx_t edgecut; |
---|
11 | idx_t options[]={0}, wgtflag=0, numflag=0, ncon=1; |
---|
12 | real_t tpwgts[2*nparts], ubvec[]={1.05}; |
---|
13 | |
---|
14 | /* int ParMETIS V3 PartKway ( |
---|
15 | idx t *vtxdist, idx t *xadj, idx t *adjncy, idx t *vwgt, idx t *adjwgt, |
---|
16 | idx t *wgtflag, idx t *numflag, idx t *ncon, idx t *nparts, |
---|
17 | real t *tpwgts, real t *ubvec, idx t *options, idx t *edgecut, idx t *part, |
---|
18 | MPI Comm *comm ) |
---|
19 | */ |
---|
20 | |
---|
21 | /* printf("sizeof : %d %d \n", sizeof(int), sizeof(idx_t)); |
---|
22 | printf("partition_graph : %d %d %d\n", mpi_rank, mpi_size, nparts); |
---|
23 | printf("vtxdist : \n"); |
---|
24 | for(int i=0; i<mpi_size+1; i++) printf("%d %d %d\n", mpi_rank, i, vtxdist[i]); */ |
---|
25 | |
---|
26 | for(int i=0; i<2*nparts; i++) tpwgts[i] = 1./nparts; |
---|
27 | ParMETIS_V3_PartKway(vtxdist, xadj_loc, adjncy_loc, NULL, NULL, |
---|
28 | &wgtflag, &numflag, &ncon, &nparts, |
---|
29 | tpwgts, ubvec, options, &edgecut, part, &comm); |
---|
30 | } |
---|