Lines Matching refs:nparts
33 PetscInt m, M, nparts, *indices, r, d, *parts, i, start, end, loc; in MatPartitioningApply_Average() local
38 nparts = part->n; in MatPartitioningApply_Average()
39 PetscCall(PetscMalloc1(nparts, &parts)); in MatPartitioningApply_Average()
40 d = M / nparts; in MatPartitioningApply_Average()
41 for (i = 0; i < nparts; i++) parts[i] = d; in MatPartitioningApply_Average()
42 r = M % nparts; in MatPartitioningApply_Average()
44 for (i = 1; i < nparts; i++) parts[i] += parts[i - 1]; in MatPartitioningApply_Average()
48 PetscCall(PetscFindInt(i, nparts, parts, &loc)); in MatPartitioningApply_Average()
378 PetscMPIInt nparts; in MatPartitioningViewImbalance() local
386 PetscCall(PetscMPIIntCast(matp->n, &nparts)); in MatPartitioningViewImbalance()
387 PetscCall(PetscCalloc2(nparts, &subdomainsizes, nparts, &subdomainsizes_tmp)); in MatPartitioningViewImbalance()
391 …PetscCallMPI(MPIU_Allreduce(subdomainsizes_tmp, subdomainsizes, nparts, MPIU_INT, MPI_SUM, PetscOb… in MatPartitioningViewImbalance()
394 for (PetscMPIInt i = 0; i < nparts; i++) { in MatPartitioningViewImbalance()
399 avgsub /= nparts; in MatPartitioningViewImbalance()