Home
last modified time | relevance | path

Searched refs:nnz2 (Results 1 – 8 of 8) 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()
255 make_coo_pair(vcu->jmap2_d, vmpi->jmap2, vmpi->nnz2 + 1), in SetPreallocationCOO()
282 PETSC_KERNEL_DECL void add_remote_coo_values(const PetscScalar *PETSC_RESTRICT vv, PetscCount nnz2,… 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()
359 …if (const auto nnz2 = vmpi->nnz2) PetscCall(PetscCUPMLaunchKernel1D(nnz2, 0, stream, kernels::add_… in SetValuesCOO() local
/petsc/src/mat/impls/sbaij/seq/
H A Drelax.h12 PetscInt nnz2 = nnz, rem = nnz & 0x3; \
20 nnz2 -= rem; \
22 while (nnz2 > 0) { \
26 nnz2 -= 4; \
/petsc/src/mat/impls/aij/seq/
H A Daij.h508 PetscInt nnz2 = nnz, rem = nnz & 0x3; \
516 nnz2 -= rem; \
518 while (nnz2 > 0) { \
522 nnz2 -= 4; \
574 PetscInt nnz2 = nnz, rem = nnz & 0x3; \
582 nnz2 -= rem; \
584 while (nnz2 > 0) { \
588 nnz2 -= 4; \
/petsc/src/vec/vec/impls/mpi/
H A Dpdvec.c1058 PetscCount *jmap2, nnz2; in VecSetPreallocationCOO_MPI() local
1067 nnz2 = n2; in VecSetPreallocationCOO_MPI()
1069 if (i2[k] == i2[k - 1]) nnz2--; in VecSetPreallocationCOO_MPI()
1073 PetscCall(PetscMalloc4(nnz2, &imap2, nnz2 + 1, &jmap2, sendlen, &sendbuf, recvlen, &recvbuf)); in VecSetPreallocationCOO_MPI()
1089 for (k = 0; k < nnz2; k++) jmap2[k + 1] += jmap2[k]; in VecSetPreallocationCOO_MPI()
1097 vmpi->nnz2 = nnz2; in VecSetPreallocationCOO_MPI()
1122 const PetscCount nnz2 = vmpi->nnz2; in VecSetValuesCOO_MPI() local
1142 for (PetscInt i = 0; i < nnz2; i++) { in VecSetValuesCOO_MPI()
H A Dpvecimpl.h51 PetscCount nnz2; /* Unique entries in recvbuf */ member
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckokkosimpl.hpp113 …mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(vecmpi->imap2, vecmpi->nnz2))); in SetUpCOO()
114 …or_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(vecmpi->jmap2, vecmpi->nnz2 + 1))); in SetUpCOO()
/petsc/src/mat/impls/aij/seq/bas/
H A Dspbas.c148 PetscInt nnz2 = irow_in[i2 + 1] - irow_in[i2]; in spbas_row_order_icol() local
152 if (nnz1 < nnz2) return -1; in spbas_row_order_icol()
153 if (nnz1 > nnz2) return 1; in spbas_row_order_icol()
/petsc/src/vec/vec/impls/mpi/kokkos/
H A Dmpikok.kokkos.cxx194 …Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, vecmpi->nnz2), KOKKOS_LAMBDA(PetscCount i… in VecSetValuesCOO_MPIKokkos()