impls/parmetis/partparmetis.c

93   real_t                    *tpwgts;                  /* The fraction of vertex weights assigned to each partition */
105   PetscCall(PetscMalloc4(size + 1, &vtxdist, nparts * ncon, &tpwgts, ncon, &ubvec, nvtxs, &assignment));
114     PetscCall(PetscFree4(vtxdist, tpwgts, ubvec, assignment));
128       tpwgts[p] = tpd;
132         tpwgts[p] = (real_t)PetscMax(tpwgts[p], PETSC_SMALL);
133         sumt += tpwgts[p];
135       for (p = 0; p < nparts; ++p) tpwgts[p] /= sumt;
136       for (p = 0, sumt = 0.0; p < nparts - 1; ++p) sumt += tpwgts[p];
137       tpwgts[nparts - 1] = (real_t)(1. - sumt);
140     for (p = 0; p < nparts; ++p) tpwgts[p] = (real_t)(1.0 / nparts);
167         err = METIS_PartGraphRecursive(&nvtxs, &ncon, xadj, adjncy, vwgt, NULL, adjwgt, &nparts, tpwgts, ubvec, options, &part->edgeCut, assignment);
179         err = METIS_PartGraphKway(&nvtxs, &ncon, xadj, adjncy, vwgt, NULL, adjwgt, &nparts, tpwgts, ubvec, options, &part->edgeCut, assignment);
205       err = ParMETIS_V3_PartKway(vtxdist, xadj, adjncy, vwgt, adjwgt, &wgtflag, &numflag, &ncon, &nparts, tpwgts, ubvec, options, &part->edgeCut, assignment, &pcomm);
222   PetscCall(PetscFree4(vtxdist, tpwgts, ubvec, assignment));