Home
last modified time | relevance | path

Searched refs:jmap2 (Results 1 – 6 of 6) sorted by relevance

/petsc/src/vec/vec/impls/mpi/
H A Dpdvec.c19 PetscCall(PetscFree4(vmpi->imap2, vmpi->jmap2, vmpi->sendbuf, vmpi->recvbuf)); in VecResetPreallocationCOO_MPI()
1058 PetscCount *jmap2, nnz2; in VecSetPreallocationCOO_MPI() local
1073 PetscCall(PetscMalloc4(nnz2, &imap2, nnz2 + 1, &jmap2, sendlen, &sendbuf, recvlen, &recvbuf)); in VecSetPreallocationCOO_MPI()
1076 jmap2[0] = 0; in VecSetPreallocationCOO_MPI()
1077 jmap2++; in VecSetPreallocationCOO_MPI()
1082 jmap2[p] = 1; in VecSetPreallocationCOO_MPI()
1085 jmap2[p]++; in VecSetPreallocationCOO_MPI()
1088 jmap2--; in VecSetPreallocationCOO_MPI()
1089 for (k = 0; k < nnz2; k++) jmap2[k + 1] += jmap2[k]; in VecSetPreallocationCOO_MPI()
1099 vmpi->jmap2 = jmap2; in VecSetPreallocationCOO_MPI()
[all …]
H A Dpvecimpl.h53 PetscCount *jmap2; /* [nnz2+1] */ member
/petsc/src/vec/vec/impls/mpi/cupm/
H A Dvecmpicupm_impl.hpp255 make_coo_pair(vcu->jmap2_d, vmpi->jmap2, vmpi->nnz2 + 1), in SetPreallocationCOO()
282 … const PetscCount *PETSC_RESTRICT imap2, const PetscCount *PETSC_RESTRICT jmap2, const PetscCount … in add_remote_coo_values() argument
284 …add_coo_values_impl(vv, nnz2, jmap2, perm2, ADD_VALUES, xv, [=](PetscCount i) { return imap2[i]; }… in add_remote_coo_values()
/petsc/src/vec/vec/impls/mpi/kokkos/
H A Dmpikok.kokkos.cxx157 const PetscCountKokkosView &jmap2 = veckok->jmap2_d; in VecSetValuesCOO_MPIKokkos() local
195 for (PetscCount k = jmap2(i); k < jmap2(i + 1); k++) xv(imap2(i)) += recvbuf(perm2(k)); in VecSetValuesCOO_MPIKokkos()
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckokkosimpl.hpp114 …rror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(vecmpi->jmap2, vecmpi->nnz2 + 1)… in SetUpCOO()
/petsc/src/mat/impls/aij/mpi/
H A Dmpiaij.c6081 …], const PetscCount rowEnd2[], const PetscCount jmap1[], const PetscCount jmap2[], PetscCount imap… in MatMergeEntries_Internal() argument
6101 b2 += jmap2[t2 + 1] - jmap2[t2]; /* Jump to next unique remote nonzero */ in MatMergeEntries_Internal()
6114 b2 += jmap2[t2 + 1] - jmap2[t2]; in MatMergeEntries_Internal()
6130 b2 += jmap2[t2 + 1] - jmap2[t2]; in MatMergeEntries_Internal()