Searched refs:sendRowLen (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 398 PetscInt *sendRowLen, *recvRowLen; // lengths of rows of I need to send/recv per process in MatMPIAIJKokkosReduceBegin() local
400 …PetscCall(PetscMalloc5(sendRowCnt, &sendRowLen, recvRowCnt + 1, &recvRowLen, nranks, &sdisp, niran… in MatMPIAIJKokkosReduceBegin()
402 for (PetscInt i = 0; i < sendRowCnt; i++) sendRowLen[i] = E_RowLen[rmine[i]]; in MatMPIAIJKokkosReduceBegin()
406 …for (PetscInt i = 0; i < nranks; i++) PetscCallMPI(MPIU_Isend(&sendRowLen[roffset[i]], roffset[i +… in MatMPIAIJKokkosReduceBegin()
592 PetscCall(PetscFree5(sendRowLen, recvRowLen, sdisp, rdisp, reqs)); in MatMPIAIJKokkosReduceBegin()
|