Searched refs:lsizes (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/vec/is/is/utils/ |
| H A D | iscoloring.c | 519 PetscInt n, *starts = NULL, *sums = NULL, *lsizes = NULL, *newi = NULL; in ISPartitioningToNumbering() local
548 PetscCall(PetscMalloc3(np, &lsizes, np, &starts, np, &sums)); in ISPartitioningToNumbering()
549 PetscCall(PetscArrayzero(lsizes, np)); in ISPartitioningToNumbering()
550 for (PetscInt i = 0; i < n; i++) lsizes[indices[i]]++; in ISPartitioningToNumbering()
551 PetscCallMPI(MPIU_Allreduce(lsizes, sums, np, MPIU_INT, MPI_SUM, comm)); in ISPartitioningToNumbering()
552 PetscCallMPI(MPI_Scan(lsizes, starts, np, MPIU_INT, MPI_SUM, comm)); in ISPartitioningToNumbering()
553 for (PetscMPIInt i = 0; i < np; i++) starts[i] -= lsizes[i]; in ISPartitioningToNumbering()
564 PetscCall(PetscFree3(lsizes, starts, sums)); in ISPartitioningToNumbering()
604 PetscInt i, n, *lsizes; in ISPartitioningCount() local
632 PetscCall(PetscCalloc1(len, &lsizes)); in ISPartitioningCount()
[all …]
|
| /petsc/src/binding/petsc4py/test/ |
| H A D | test_dmda.py | 47 starts, lsizes = self.da.getCorners()
50 self.assertEqual(dim, len(lsizes))
54 self.assertEqual(e - s, lsizes[i])
59 starts, lsizes = self.da.getGhostCorners()
62 self.assertEqual(dim, len(lsizes))
66 self.assertEqual(e - s, lsizes[i])
|
| /petsc/src/dm/impls/da/ |
| H A D | gr2.c | 581 PetscMPIInt gsizes[4], lsizes[4], lstarts[4], asiz, dof; in DMDAArrayMPIIO() local
613 lsizes[0] = dof; in DMDAArrayMPIIO()
614 PetscCall(PetscMPIIntCast((dd->xe - dd->xs) / dof, lsizes + 1)); in DMDAArrayMPIIO()
615 PetscCall(PetscMPIIntCast(dd->ye - dd->ys, lsizes + 2)); in DMDAArrayMPIIO()
616 PetscCall(PetscMPIIntCast(dd->ze - dd->zs, lsizes + 3)); in DMDAArrayMPIIO()
621 …PetscCallMPI(MPI_Type_create_subarray((PetscMPIInt)(da->dim + 1), gsizes, lsizes, lstarts, MPI_ORD… in DMDAArrayMPIIO()
628 asiz = lsizes[1] * (lsizes[2] > 0 ? lsizes[2] : 1) * (lsizes[3] > 0 ? lsizes[3] : 1) * dof; in DMDAArrayMPIIO()
|