Lines Matching refs:nvtxs
206 PetscInt nvtxs = numVertices; /* The number of vertices in full graph */ in PetscPartitionerPartition_PTScotch() local
221 PetscCall(PetscMalloc2(size + 1, &vtxdist, PetscMax(nvtxs, 1), &assignment)); in PetscPartitionerPartition_PTScotch()
224 PetscCallMPI(MPI_Allgather(&nvtxs, 1, MPIU_INT, &vtxdist[1], 1, MPIU_INT, comm)); in PetscPartitionerPartition_PTScotch()
238 PetscCall(PetscMalloc1(nvtxs, &vwgt)); in PetscPartitionerPartition_PTScotch()
239 for (v = 0; v < nvtxs; ++v) PetscCall(PetscSectionGetDof(vertSection, v, &vwgt[v])); in PetscPartitionerPartition_PTScotch()
243 PetscCall(PetscMalloc1(xadj[nvtxs], &adjwgt)); in PetscPartitionerPartition_PTScotch()
244 …for (PetscInt e = 0; e < xadj[nvtxs]; ++e) PetscCall(PetscSectionGetDof(edgeSection, e, &adjwgt[e]… in PetscPartitionerPartition_PTScotch()
266 …if (rank == p) PetscCall(PTScotch_PartGraph_Seq(strat, imbal, nvtxs, xadj, adjncy, vwgt, adjwgt, n… in PetscPartitionerPartition_PTScotch()
273 PetscCallMPI(MPI_Comm_split(pts->pcomm, !!nvtxs, rank, &pcomm)); in PetscPartitionerPartition_PTScotch()
281 …if (nvtxs) PetscCall(PTScotch_PartGraph_MPI(strat, imbal, vtxdist, xadj, adjncy, vwgt, adjwgt, npa… in PetscPartitionerPartition_PTScotch()
290 for (v = 0; v < nvtxs; ++v) PetscCall(PetscSectionAddDof(partSection, assignment[v], 1)); in PetscPartitionerPartition_PTScotch()
291 PetscCall(PetscMalloc1(nvtxs, &points)); in PetscPartitionerPartition_PTScotch()
293 for (v = 0; v < nvtxs; ++v) { in PetscPartitionerPartition_PTScotch()
297 …PetscCheck(i == nvtxs, comm, PETSC_ERR_PLIB, "Number of points %" PetscInt_FMT " should be %" Pets… in PetscPartitionerPartition_PTScotch()
298 PetscCall(ISCreateGeneral(comm, nvtxs, points, PETSC_OWN_POINTER, partition)); in PetscPartitionerPartition_PTScotch()