Searched refs:vsend (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 177 const auto &vsend = coo->sendbuf; in MatSetValuesCOO_MPIAIJHIPSPARSE() local
208 …(coo->sendlen + 255) / 256), dim3(256), 0, PetscDefaultHipStream, v1, coo->sendlen, Cperm1, vsend); in MatSetValuesCOO_MPIAIJHIPSPARSE()
213 …PetscCall(PetscSFReduceWithMemTypeBegin(coo->sf, MPIU_SCALAR, PETSC_MEMTYPE_HIP, vsend, PETSC_MEMT… in MatSetValuesCOO_MPIAIJHIPSPARSE()
219 PetscCall(PetscSFReduceEnd(coo->sf, MPIU_SCALAR, vsend, v2, MPI_REPLACE)); in MatSetValuesCOO_MPIAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 177 const auto &vsend = coo->sendbuf; in MatSetValuesCOO_MPIAIJCUSPARSE() local
208 MatPackCOOValues<<<(coo->sendlen + 255) / 256, 256>>>(v1, coo->sendlen, Cperm1, vsend); in MatSetValuesCOO_MPIAIJCUSPARSE()
213 …PetscCall(PetscSFReduceWithMemTypeBegin(coo->sf, MPIU_SCALAR, PETSC_MEMTYPE_CUDA, vsend, PETSC_MEM… in MatSetValuesCOO_MPIAIJCUSPARSE()
219 PetscCall(PetscSFReduceEnd(coo->sf, MPIU_SCALAR, vsend, v2, MPI_REPLACE)); in MatSetValuesCOO_MPIAIJCUSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 1597 const auto &vsend = coo->sendbuf; in MatSetValuesCOO_MPIAIJKokkos() local
1628 …s::parallel_for(Kokkos::RangePolicy<>(exec, 0, vsend.extent(0)), KOKKOS_LAMBDA(const PetscCount i)… in MatSetValuesCOO_MPIAIJKokkos()
1631 …PetscCall(PetscSFReduceWithMemTypeBegin(coo->sf, MPIU_SCALAR, PETSC_MEMTYPE_KOKKOS, vsend.data(), … in MatSetValuesCOO_MPIAIJKokkos()
1645 PetscCall(PetscSFReduceEnd(coo->sf, MPIU_SCALAR, vsend.data(), v2.data(), MPI_REPLACE)); in MatSetValuesCOO_MPIAIJKokkos()
|