Searched refs:rdisp (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 389 PetscInt *sdisp, *rdisp; in MatMPIAIJKokkosReduceBegin() local
400 …, &sendRowLen, recvRowCnt + 1, &recvRowLen, nranks, &sdisp, niranks + 1, &rdisp, nranks + niranks,… in MatMPIAIJKokkosReduceBegin()
410 rdisp[0] = 0; in MatMPIAIJKokkosReduceBegin()
412 rdisp[i + 1] = rdisp[i]; in MatMPIAIJKokkosReduceBegin()
413 for (PetscInt j = ioffset[i]; j < ioffset[i + 1]; j++) rdisp[i + 1] += recvRowLen[j]; in MatMPIAIJKokkosReduceBegin()
419 …for (PetscInt i = 0; i < niranks; i++) PetscCallMPI(MPIU_Isend(&rdisp[i], 1, MPIU_INT, iranks[i], … in MatMPIAIJKokkosReduceBegin()
423 PetscInt nroots = rdisp[niranks]; // roots are nonzeros I will recv in MatMPIAIJKokkosReduceBegin()
592 PetscCall(PetscFree5(sendRowLen, recvRowLen, sdisp, rdisp, reqs)); in MatMPIAIJKokkosReduceBegin()
795 PetscInt *sdisp, *rdisp; in MatMPIAIJKokkosBcastBegin() local
801 PetscCall(PetscMalloc3(niranks + 1, &sdisp, nranks, &rdisp, niranks + nranks, &reqs)); in MatMPIAIJKokkosBcastBegin()
[all …]
|