Searched refs:leafBuf (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 124 MatScalarKokkosView rootBuf, leafBuf; member
599 mm->leafBuf = MatScalarKokkosView(NoInit("leafBuf"), nleaves); in MatMPIAIJKokkosReduceBegin()
636 auto &leafBuf = mm->leafBuf; in MatMPIAIJKokkosReduceBegin() local
657 MatScalar &val = leafBuf(disp + j); in MatMPIAIJKokkosReduceBegin()
669 …PetscCall(PetscSFReduceWithMemTypeBegin(reduceSF, MPIU_SCALAR, PETSC_MEMTYPE_KOKKOS, leafBuf.data(… in MatMPIAIJKokkosReduceBegin()
676 auto &leafBuf = mm->leafBuf; in MatMPIAIJKokkosReduceEnd() local
689 PetscCall(PetscSFReduceEnd(reduceSF, MPIU_SCALAR, leafBuf.data(), rootBuf.data(), MPI_REPLACE)); in MatMPIAIJKokkosReduceEnd()
919 mm->leafBuf = MatScalarKokkosView(NoInit("leafBuf"), nleaves); in MatMPIAIJKokkosBcastBegin()
963 MatScalarKokkosView &leafBuf = mm->leafBuf; in MatMPIAIJKokkosBcastBegin() local
996 …MPIU_SCALAR, PETSC_MEMTYPE_KOKKOS, rootBuf.data(), PETSC_MEMTYPE_KOKKOS, leafBuf.data(), MPI_REPLA… in MatMPIAIJKokkosBcastBegin()
[all …]
|