Lines Matching refs:sizes

96 …PetscInt           nmax, *sindices, *rindices, idx, lastidx, *sindices2, *rindices2, *sizes, *star…  in AOMap_MemoryScalable_private()  local
113 PetscCall(PetscCalloc2(2 * size, &sizes, n, &owner)); in AOMap_MemoryScalable_private()
126 sizes[2 * j]++; /* num of indices to be sent */ in AOMap_MemoryScalable_private()
127 sizes[2 * j + 1] = 1; /* send to proc[j] */ in AOMap_MemoryScalable_private()
134 sizes[2 * rank] = sizes[2 * rank + 1] = 0; /* do not receive from self! */ in AOMap_MemoryScalable_private()
136 for (PetscMPIInt i = 0; i < size; i++) nsends += sizes[2 * i + 1]; in AOMap_MemoryScalable_private()
139 PetscCall(PetscMaxSum(comm, sizes, &nmax, &inreceives)); in AOMap_MemoryScalable_private()
164 for (PetscMPIInt i = 1; i < size; i++) start[i] = start[i - 1] + sizes[2 * i - 2]; in AOMap_MemoryScalable_private()
178 for (PetscMPIInt i = 1; i < size; i++) start[i] = start[i - 1] + sizes[2 * i - 2]; in AOMap_MemoryScalable_private()
181 if (sizes[2 * i + 1]) { in AOMap_MemoryScalable_private()
183 …PetscCallMPI(MPIU_Isend(sindices + start[i], sizes[2 * i], MPIU_INT, i, tag1, comm, send_waits + c… in AOMap_MemoryScalable_private()
185 …PetscCallMPI(MPIU_Irecv(sindices2 + start[i], sizes[2 * i], MPIU_INT, i, tag2, comm, recv_waits2 +… in AOMap_MemoryScalable_private()
227 PetscCall(PetscFree2(sizes, owner)); in AOMap_MemoryScalable_private()
272 PetscInt *owner, *start, *sizes, nsends, nreceives; in AOCreateMemoryScalable_private() local
291 PetscCall(PetscCalloc1(2 * size, &sizes)); in AOCreateMemoryScalable_private()
302 sizes[2 * j] += 2; /* num of indices to be sent - in pairs (ip,ia) */ in AOCreateMemoryScalable_private()
303 sizes[2 * j + 1] = 1; /* send to proc[j] */ in AOCreateMemoryScalable_private()
309 sizes[2 * rank] = sizes[2 * rank + 1] = 0; /* do not receive from self! */ in AOCreateMemoryScalable_private()
311 for (i = 0; i < size; i++) nsends += sizes[2 * i + 1]; in AOCreateMemoryScalable_private()
314 PetscCall(PetscMaxSum(comm, sizes, &nmax, &nreceives)); in AOCreateMemoryScalable_private()
330 for (i = 1; i < size; i++) start[i] = start[i - 1] + sizes[2 * i - 2]; in AOCreateMemoryScalable_private()
346 for (PetscMPIInt i = 1; i < size; i++) start[i] = start[i - 1] + sizes[2 * i - 2]; in AOCreateMemoryScalable_private()
349 if (sizes[2 * i + 1]) { in AOCreateMemoryScalable_private()
350 …PetscCallMPI(MPIU_Isend(sindices + start[i], sizes[2 * i], MPIU_INT, i, tag, comm, send_waits + co… in AOCreateMemoryScalable_private()
381 PetscCall(PetscFree(sizes)); in AOCreateMemoryScalable_private()