mpi/kokkos/mpiaijkok.kokkos.cxx

621     PetscInt teamSize, vectorLength, rowsPerTeam;
623     teamSize = vectorLength = rowsPerTeam = -1;
624     PetscCall(MatMergeGetLaunchParameters<DefaultExecutionSpace>(Em, Enz, -1, teamSize, vectorLength, rowsPerTeam));
625     mm->E_TeamSize     = teamSize;
640   PetscInt    teamSize     = mm->E_TeamSize;
647     Kokkos::TeamPolicy<>(PetscGetKokkosExecutionSpace(), workSets, teamSize, vectorLength), KOKKOS_LAMBDA(const KokkosTeamMemberType &t) {
934     PetscInt teamSize, vectorLength, rowsPerTeam;
936     teamSize = vectorLength = rowsPerTeam = -1;
937     PetscCall(MatMergeGetLaunchParameters<DefaultExecutionSpace>(mm->irootloc.extent(0), mm->rootBuf.extent(0), -1, teamSize, vectorLength, rowsPerTeam));
938     mm->E_TeamSize     = teamSize;
942     teamSize = vectorLength = rowsPerTeam = -1;
943     PetscCall(MatMergeGetLaunchParameters<DefaultExecutionSpace>(Fm, Fnz, -1, teamSize, vectorLength, rowsPerTeam));
944     mm->F_TeamSize     = teamSize;
967   PetscInt teamSize     = mm->E_TeamSize;
974     Kokkos::TeamPolicy<>(PetscGetKokkosExecutionSpace(), workSets, teamSize, vectorLength), KOKKOS_LAMBDA(const KokkosTeamMemberType &t) {
1016   PetscInt             teamSize     = mm->F_TeamSize;
1025     Kokkos::TeamPolicy<>(PetscGetKokkosExecutionSpace(), workSets, teamSize, vectorLength), KOKKOS_LAMBDA(const KokkosTeamMemberType &t) {