Home
last modified time | relevance | path

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

/petsc/src/vec/vec/impls/mpi/cupm/
H A Dvecmpicupm_impl.hpp254 make_coo_pair(vcu->imap2_d, vmpi->imap2, vmpi->nnz2), in SetPreallocationCOO()
282 …lar *PETSC_RESTRICT vv, PetscCount nnz2, const PetscCount *PETSC_RESTRICT imap2, 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/
H A Dpdvec.c19 PetscCall(PetscFree4(vmpi->imap2, vmpi->jmap2, vmpi->sendbuf, vmpi->recvbuf)); in VecResetPreallocationCOO_MPI()
1057 PetscCount *imap2; in VecSetPreallocationCOO_MPI() local
1073 PetscCall(PetscMalloc4(nnz2, &imap2, nnz2 + 1, &jmap2, sendlen, &sendbuf, recvlen, &recvbuf)); in VecSetPreallocationCOO_MPI()
1081 imap2[p] = i2[k] - rstart; in VecSetPreallocationCOO_MPI()
1098 vmpi->imap2 = imap2; in VecSetPreallocationCOO_MPI()
1118 const PetscCount *imap2 = vmpi->imap2; in VecSetValuesCOO_MPI() local
1143 for (PetscCount k = jmap2[i]; k < jmap2[i + 1]; k++) a[imap2[i]] += recvbuf[perm2[k]]; in VecSetValuesCOO_MPI()
H A Dpvecimpl.h52 PetscCount *imap2; /* [nnz2]: i-th unique entry in recvbuf is imap2[i]-th entry in the vector */ member
/petsc/src/vec/vec/impls/mpi/kokkos/
H A Dmpikok.kokkos.cxx156 const PetscCountKokkosView &imap2 = veckok->imap2_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.hpp113 …mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(vecmpi->imap2, vecmpi->nnz2))); in SetUpCOO()
/petsc/src/mat/impls/aij/mpi/
H A Dmpiaij.c6081 …scCount jmap1[], const PetscCount jmap2[], PetscCount imap1[], PetscCount imap2[], PetscInt i[], P… in MatMergeEntries_Internal() argument
6099 imap2[t2] = t; in MatMergeEntries_Internal()
6113 imap2[t2] = t; in MatMergeEntries_Internal()
6129 imap2[t2] = t; in MatMergeEntries_Internal()