Home
last modified time | relevance | path

Searched refs:recvlen (Results 1 – 9 of 9) sorted by relevance

/petsc/src/vec/vec/impls/seq/kokkos/ !
H A Dveckokkosimpl.hpp115 …ror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(vecmpi->perm2, vecmpi->recvlen))); in SetUpCOO()
118 …_view_and_copy(DefaultMemorySpace(), PetscScalarKokkosViewHost(vecmpi->recvbuf, vecmpi->recvlen))); in SetUpCOO()
/petsc/src/vec/vec/impls/mpi/cupm/ !
H A Dvecmpicupm_impl.hpp256 make_coo_pair(vcu->perm2_d, vmpi->perm2, vmpi->recvlen), in SetPreallocationCOO()
261 make_coo_pair(vcu->recvbuf_d, vmpi->recvbuf, vmpi->recvlen) in SetPreallocationCOO()
/petsc/src/vec/vec/impls/mpi/ !
H A Dpvecimpl.h57 PetscCount sendlen, recvlen; /* Lengths (in unit of PetscScalar) of send/recvbuf */ member
H A Dpdvec.c1061 PetscCount sendlen = n1 - rem, recvlen = n2; in VecSetPreallocationCOO_MPI() local
1073 PetscCall(PetscMalloc4(nnz2, &imap2, nnz2 + 1, &jmap2, sendlen, &sendbuf, recvlen, &recvbuf)); in VecSetPreallocationCOO_MPI()
1106 vmpi->recvlen = recvlen; in VecSetPreallocationCOO_MPI()
/petsc/src/mat/impls/aij/mpi/ !
H A Dmpiaij.h88 PetscInt sendlen, recvlen; /* Lengths (in unit of PetscScalar) of send/recvbuf */ member
H A Dmpiaij.c6675 coo->recvlen = nroots; in MatSetPreallocationCOO_MPIAIJ()
6696 PetscCall(PetscMalloc2(coo->sendlen, &coo->sendbuf, coo->recvlen, &coo->recvbuf)); in MatSetPreallocationCOO_MPIAIJ()
/petsc/src/mat/impls/aij/mpi/mpihipsparse/ !
H A Dmpiaijhipsparse.hip.cxx101 PetscCallHIP(hipMalloc((void **)&coo_d->recvbuf, coo_h->recvlen * sizeof(PetscScalar))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
/petsc/src/mat/impls/aij/mpi/mpicusparse/ !
H A Dmpiaijcusparse.cu101 PetscCallCUDA(cudaMalloc((void **)&coo_d->recvbuf, coo_h->recvlen * sizeof(PetscScalar))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
/petsc/src/mat/impls/aij/mpi/kokkos/ !
H A Dmpiaijkok.kokkos.cxx1536 …r_view(Kokkos::WithoutInitializing, exec, MatScalarKokkosViewHost(coo_h->recvbuf, coo_h->recvlen)); in MatCOOStruct_MPIAIJKokkos()