mpi/kokkos/mpiaijkok.kokkos.cxx

124   MatScalarKokkosView rootBuf, leafBuf;
599     mm->leafBuf  = MatScalarKokkosView(NoInit("leafBuf"), nleaves);
636   auto       &leafBuf      = mm->leafBuf;
645   // Copy rows in A/B of E to leafBuf, then pass it to rootBuf
657             MatScalar &val = leafBuf(disp + j);
669   PetscCall(PetscSFReduceWithMemTypeBegin(reduceSF, MPIU_SCALAR, PETSC_MEMTYPE_KOKKOS, leafBuf.data(), PETSC_MEMTYPE_KOKKOS, rootBuf.data(), MPI_REPLACE));
676   auto       &leafBuf  = mm->leafBuf;
689   PetscCall(PetscSFReduceEnd(reduceSF, MPIU_SCALAR, leafBuf.data(), rootBuf.data(), MPI_REPLACE));
919     mm->leafBuf   = MatScalarKokkosView(NoInit("leafBuf"), nleaves);
963   MatScalarKokkosView &leafBuf   = mm->leafBuf;
972   // Copy rows in A/B of E to rootBuf, then bcast it to leafBuf
996   PetscCall(PetscSFBcastWithMemTypeBegin(bcastSF, MPIU_SCALAR, PETSC_MEMTYPE_KOKKOS, rootBuf.data(), PETSC_MEMTYPE_KOKKOS, leafBuf.data(), MPI_REPLACE));
1015   MatScalarKokkosView &leafBuf      = mm->leafBuf;
1021   PetscCall(PetscSFBcastEnd(bcastSF, MPIU_SCALAR, rootBuf.data(), leafBuf.data(), MPI_REPLACE));
1023   // Update Fda and Foa with new data in leafBuf (as if it is Fa)
1035             PetscScalar val = leafBuf(Fii + j);