Searched refs:sdisp (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 …PetscCall(PetscMalloc5(sendRowCnt, &sendRowLen, recvRowCnt + 1, &recvRowLen, nranks, &sdisp, niran… in MatMPIAIJKokkosReduceBegin()
418 …for (PetscInt i = 0; i < nranks; i++) PetscCallMPI(MPIU_Irecv(&sdisp[i], 1, MPIU_INT, ranks[i], ta… in MatMPIAIJKokkosReduceBegin()
435 iremote[nleaves + j].index = sdisp[i] + j; 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()
803 sdisp[0] = 0; // send displacement in MatMPIAIJKokkosBcastBegin()
805 sdisp[i + 1] = sdisp[i]; in MatMPIAIJKokkosBcastBegin()
808 sdisp[i + 1] += E_RowLen[r]; in MatMPIAIJKokkosBcastBegin()
[all …]
|