Searched refs:rootBuf (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
600 mm->rootBuf = MatScalarKokkosView(NoInit("rootBuf"), nroots); in MatMPIAIJKokkosReduceBegin()
637 auto &rootBuf = mm->rootBuf; in MatMPIAIJKokkosReduceBegin() local
669 …MPIU_SCALAR, PETSC_MEMTYPE_KOKKOS, leafBuf.data(), PETSC_MEMTYPE_KOKKOS, rootBuf.data(), MPI_REPLA… in MatMPIAIJKokkosReduceBegin()
677 auto &rootBuf = mm->rootBuf; in MatMPIAIJKokkosReduceEnd() local
689 PetscCall(PetscSFReduceEnd(reduceSF, MPIU_SCALAR, leafBuf.data(), rootBuf.data(), MPI_REPLACE)); in MatMPIAIJKokkosReduceEnd()
695 for (MatRowMapType k = Fdjmap(i); k < Fdjmap(i + 1); k++) sum += rootBuf(Fdjperm(k)); in MatMPIAIJKokkosReduceEnd()
702 for (MatRowMapType k = Fojmap(i); k < Fojmap(i + 1); k++) sum += rootBuf(Fojperm(k)); in MatMPIAIJKokkosReduceEnd()
918 mm->rootBuf = MatScalarKokkosView(NoInit("rootBuf"), nroots); in MatMPIAIJKokkosBcastBegin()
937 …geGetLaunchParameters<DefaultExecutionSpace>(mm->irootloc.extent(0), mm->rootBuf.extent(0), -1, te… in MatMPIAIJKokkosBcastBegin()
[all …]
|