Searched refs:E_RowLen (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 361 PetscInt *E_NzLeft = E_NzLeft_h.data(), *E_RowLen = E_RowLen_h.data(); in MatMPIAIJKokkosReduceBegin() local
379 E_RowLen[i] = (Ai[i + 1] - Ai[i]) + (Bi[i + 1] - Bi[i]); in MatMPIAIJKokkosReduceBegin()
402 for (PetscInt i = 0; i < sendRowCnt; i++) sendRowLen[i] = E_RowLen[rmine[i]]; in MatMPIAIJKokkosReduceBegin()
426 for (PetscInt i = 0; i < Em; i++) Enz += E_RowLen[i]; in MatMPIAIJKokkosReduceBegin()
432 for (PetscInt j = roffset[i]; j < roffset[i + 1]; j++) count += E_RowLen[rmine[j]]; in MatMPIAIJKokkosReduceBegin()
760 PetscInt *E_NzLeft = E_NzLeft_h.data(), *E_RowLen = E_RowLen_h.data(); in MatMPIAIJKokkosBcastBegin() local
778 E_RowLen[i] = (Ai[i + 1] - Ai[i]) + (Bi[i + 1] - Bi[i]); in MatMPIAIJKokkosBcastBegin()
786 …PetscCall(PetscSFBcastWithMemTypeBegin(ownerSF, MPIU_INT, PETSC_MEMTYPE_HOST, E_RowLen, PETSC_MEMT… in MatMPIAIJKokkosBcastBegin()
787 PetscCall(PetscSFBcastEnd(ownerSF, MPIU_INT, E_RowLen, &Fi[1], MPI_REPLACE)); in MatMPIAIJKokkosBcastBegin()
808 sdisp[i + 1] += E_RowLen[r]; in MatMPIAIJKokkosBcastBegin()
[all …]
|