Lines Matching refs:offsets
39 PetscInt *parts_indices, i, j, mat_localsize, *offsets; in MatPartitioningApply_Hierarchical() local
95 PetscCall(PetscMalloc1(hpart->ncoarseparts + 1, &offsets)); in MatPartitioningApply_Hierarchical()
98 offsets[0] = 0; in MatPartitioningApply_Hierarchical()
99 if (part->n % hpart->nfineparts != 0) offsets[1] = part->n % hpart->nfineparts; in MatPartitioningApply_Hierarchical()
100 else offsets[1] = hpart->nfineparts; in MatPartitioningApply_Hierarchical()
102 part_weights[0] = ((PetscReal)offsets[1]) / part->n; in MatPartitioningApply_Hierarchical()
105 offsets[i] = hpart->nfineparts; in MatPartitioningApply_Hierarchical()
106 part_weights[i - 1] = ((PetscReal)offsets[i]) / part->n; in MatPartitioningApply_Hierarchical()
109 offsets[0] = 0; in MatPartitioningApply_Hierarchical()
110 for (i = 1; i <= hpart->ncoarseparts; i++) offsets[i] += offsets[i - 1]; in MatPartitioningApply_Hierarchical()
203 …PetscCall(MatPartitioningSetNParts(hpart->fineMatPart, offsets[rank + 1 + i] - offsets[rank + i])); in MatPartitioningApply_Hierarchical()
232 …for (j = 0; j < bs; j++) parts_indices[bs * i + j] = fineparts_indices[i] + offsets[coarseparts_in… in MatPartitioningApply_Hierarchical()
236 PetscCall(PetscFree(offsets)); in MatPartitioningApply_Hierarchical()