| /petsc/src/vec/vec/impls/mpi/cupm/ |
| H A D | vecmpicupm_impl.hpp | 257 make_coo_pair(vcu->Cperm_d, vmpi->Cperm, vmpi->sendlen) in SetPreallocationCOO()
260 make_coo_pair(vcu->sendbuf_d, vmpi->sendbuf, vmpi->sendlen), in SetPreallocationCOO()
345 if (const auto sendlen = vmpi->sendlen) { in SetValuesCOO() local
346 …PetscCall(PetscCUPMLaunchKernel1D(sendlen, 0, stream, kernels::pack_coo_values, vv, sendlen, vcu->… in SetValuesCOO()
|
| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 99 PetscCallHIP(hipMalloc((void **)&coo_d->Cperm1, coo_h->sendlen * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
100 PetscCallHIP(hipMalloc((void **)&coo_d->sendbuf, coo_h->sendlen * sizeof(PetscScalar))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
113 …PetscCallHIP(hipMemcpy(coo_d->Cperm1, coo_h->Cperm1, coo_h->sendlen * sizeof(PetscCount), hipMemcp… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
207 if (coo->sendlen) { in MatSetValuesCOO_MPIAIJHIPSPARSE()
208 …EL_NAME(MatPackCOOValues), dim3((coo->sendlen + 255) / 256), dim3(256), 0, PetscDefaultHipStream, … in MatSetValuesCOO_MPIAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 99 PetscCallCUDA(cudaMalloc((void **)&coo_d->Cperm1, coo_h->sendlen * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
100 PetscCallCUDA(cudaMalloc((void **)&coo_d->sendbuf, coo_h->sendlen * sizeof(PetscScalar))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
113 …PetscCallCUDA(cudaMemcpy(coo_d->Cperm1, coo_h->Cperm1, coo_h->sendlen * sizeof(PetscCount), cudaMe… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
207 if (coo->sendlen) { in MatSetValuesCOO_MPIAIJCUSPARSE()
208 MatPackCOOValues<<<(coo->sendlen + 255) / 256, 256>>>(v1, coo->sendlen, Cperm1, vsend); in MatSetValuesCOO_MPIAIJCUSPARSE()
|
| /petsc/src/vec/vec/impls/seq/kokkos/ |
| H A D | veckokkosimpl.hpp | 116 …ror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(vecmpi->Cperm, vecmpi->sendlen))); in SetUpCOO()
117 …_view_and_copy(DefaultMemorySpace(), PetscScalarKokkosViewHost(vecmpi->sendbuf, vecmpi->sendlen))); in SetUpCOO()
|
| /petsc/src/vec/vec/impls/mpi/ |
| H A D | pdvec.c | 1061 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()
1105 vmpi->sendlen = sendlen; in VecSetPreallocationCOO_MPI()
1129 for (PetscInt i = 0; i < vmpi->sendlen; i++) sendbuf[i] = v[Cperm[i]]; in VecSetValuesCOO_MPI()
|
| H A D | pvecimpl.h | 57 PetscCount sendlen, recvlen; /* Lengths (in unit of PetscScalar) of send/recvbuf */ member
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.h | 88 PetscInt sendlen, recvlen; /* Lengths (in unit of PetscScalar) of send/recvbuf */ member
|
| H A D | mpiaij.c | 6674 coo->sendlen = nleaves; in MatSetPreallocationCOO_MPIAIJ()
6696 PetscCall(PetscMalloc2(coo->sendlen, &coo->sendbuf, coo->recvlen, &coo->recvbuf)); in MatSetPreallocationCOO_MPIAIJ()
6740 for (PetscCount i = 0; i < coo->sendlen; i++) sendbuf[i] = v[Cperm1[i]]; in MatSetValuesCOO_MPIAIJ()
|
| /petsc/src/vec/vec/impls/mpi/kokkos/ |
| H A D | mpikok.kokkos.cxx | 177 …Kokkos::parallel_for(Kokkos::RangePolicy<>(PetscGetKokkosExecutionSpace(), 0, vecmpi->sendlen), KO… in VecSetValuesCOO_MPIKokkos()
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 1534 …Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Cperm1, coo_h->sendlen)); in MatCOOStruct_MPIAIJKokkos()
1535 …r_view(Kokkos::WithoutInitializing, exec, MatScalarKokkosViewHost(coo_h->sendbuf, coo_h->sendlen)); in MatCOOStruct_MPIAIJKokkos()
|