Searched refs:recvRowLen (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()
403 recvRowLen[0] = 0; // since we will make it in CSR format later in MatMPIAIJKokkosReduceBegin()
404 recvRowLen++; // advance the pointer now in MatMPIAIJKokkosReduceBegin()
405 …for (PetscInt i = 0; i < niranks; i++) PetscCallMPI(MPIU_Irecv(&recvRowLen[ioffset[i]], ioffset[i … in MatMPIAIJKokkosReduceBegin()
413 for (PetscInt j = ioffset[i]; j < ioffset[i + 1]; j++) rdisp[i + 1] += recvRowLen[j]; in MatMPIAIJKokkosReduceBegin()
415 recvRowLen--; // put it back into csr format in MatMPIAIJKokkosReduceBegin()
416 for (PetscInt i = 0; i < recvRowCnt; i++) recvRowLen[i + 1] += recvRowLen[i]; in MatMPIAIJKokkosReduceBegin()
499 PetscInt len = recvRowLen[i + 1] - recvRowLen[i]; in MatMPIAIJKokkosReduceBegin()
500 PetscCall(PetscArraycpy(jbuf2, &recvCol[recvRowLen[i]], len)); in MatMPIAIJKokkosReduceBegin()
[all …]
|