Home
last modified time | relevance | path

Searched refs:jmap1 (Results 1 – 10 of 10) sorted by relevance

/petsc/src/vec/vec/impls/seq/
H A Dbvec2.c567 PetscCall(PetscFree(vs->jmap1)); /* Destroy old stuff */ in VecResetPreallocationCOO_Seq()
577 PetscCount *perm1, *jmap1; in VecSetPreallocationCOO_Seq() local
596 PetscCall(PetscCalloc1(m + 1, &jmap1)); in VecSetPreallocationCOO_Seq()
597 for (; k < coo_n; k++) jmap1[i[k] + 1]++; /* Count repeats of each entry */ in VecSetPreallocationCOO_Seq()
598 …for (k = 0; k < m; k++) jmap1[k + 1] += jmap1[k]; /* Transform jmap[] to CSR-like data structure */ in VecSetPreallocationCOO_Seq()
612 vs->jmap1 = jmap1; /* [m+1] */ in VecSetPreallocationCOO_Seq()
620 const PetscCount *perm1 = vs->perm1, *jmap1 = vs->jmap1; in VecSetValuesCOO_Seq() local
629 for (PetscCount j = jmap1[i]; j < jmap1[i + 1]; j++) sum += coo_v[perm1[j]]; in VecSetValuesCOO_Seq()
/petsc/src/vec/vec/impls/mpi/
H A Dpdvec.c16 PetscCall(PetscFree(vmpi->jmap1)); in VecResetPreallocationCOO_MPI()
948 PetscCount tot1, *jmap1, *perm1; in VecSetPreallocationCOO_MPI() local
949 PetscCall(PetscCalloc1(m + 1, &jmap1)); in VecSetPreallocationCOO_MPI()
950 for (k = nneg; k < rem; k++) jmap1[i1[k] - rstart + 1]++; /* Count repeats of each local entry */ in VecSetPreallocationCOO_MPI()
951 …for (k = 0; k < m; k++) jmap1[k + 1] += jmap1[k]; /* Transform jmap1[] to CSR-like data st… in VecSetPreallocationCOO_MPI()
952 tot1 = jmap1[m]; in VecSetPreallocationCOO_MPI()
1095 vmpi->jmap1 = jmap1; in VecSetPreallocationCOO_MPI()
1116 const PetscCount *jmap1 = vmpi->jmap1; in VecSetValuesCOO_MPI() local
1136 for (PetscCount k = jmap1[i]; k < jmap1[i + 1]; k++) sum += v[perm1[k]]; in VecSetValuesCOO_MPI()
H A Dpvecimpl.h48 …PetscCount *jmap1; /* [m+1]: i-th entry of the vector has jmap1[i+1]-jmap1[i] repeats in COO array… member
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckokkosimpl.hpp103 …create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(vecseq->jmap1, m + 1))); in SetUpCOO()
111 …create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(vecmpi->jmap1, m + 1))); in SetUpCOO()
H A Dveckok.kokkos.cxx1578 const PetscCountKokkosView &jmap1 = veckok->jmap1_d; in VecSetValuesCOO_SeqKokkos() local
1600 for (PetscCount k = jmap1(i); k < jmap1(i + 1); k++) sum += vv(perm1(k)); in VecSetValuesCOO_SeqKokkos()
/petsc/src/vec/vec/impls/
H A Ddvecimpl.h18 …PetscCount *jmap1; /* [m+1]: perm1[jmap1[i]..jmap1[i+1]) give indices of entries in v[] associated… member
/petsc/src/vec/vec/impls/mpi/kokkos/
H A Dmpikok.kokkos.cxx154 const PetscCountKokkosView &jmap1 = veckok->jmap1_d; in VecSetValuesCOO_MPIKokkos() local
186 for (PetscCount k = jmap1(i); k < jmap1(i + 1); k++) sum += vv(perm1(k)); in VecSetValuesCOO_MPIKokkos()
/petsc/src/vec/vec/impls/seq/cupm/
H A Dvecseqcupm.hpp164 …cScalar *PETSC_RESTRICT v, PetscCount n, const PetscCount *PETSC_RESTRICT jmap1, const PetscCount … in add_coo_values() argument
166 add_coo_values_impl(v, n, jmap1, perm1, imode, xv, [](PetscCount i) { return i; }); in add_coo_values()
/petsc/src/mat/impls/aij/mpi/
H A Dmpiaij.c6081 …onst PetscCount rowBegin2[], const PetscCount rowEnd2[], const PetscCount jmap1[], const PetscCoun… in MatMergeEntries_Internal() argument
6100 b1 += jmap1[t1 + 1] - jmap1[t1]; /* Jump to next unique local nonzero */ in MatMergeEntries_Internal()
6108 b1 += jmap1[t1 + 1] - jmap1[t1]; in MatMergeEntries_Internal()
6123 b1 += jmap1[t1 + 1] - jmap1[t1]; in MatMergeEntries_Internal()
/petsc/include/petsc/private/
H A Dveccupmimpl.h1107 make_coo_pair(vcu->jmap1_d, vimpl->jmap1, v->map->n + 1), in SetPreallocationCOO_CUPMBase()