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));
627     mm->E_RowsPerTeam  = rowsPerTeam;
642   PetscInt    rowsPerTeam  = mm->E_RowsPerTeam;
643   PetscInt    workSets     = (Em + rowsPerTeam - 1) / rowsPerTeam;
648       Kokkos::parallel_for(Kokkos::TeamThreadRange(t, 0, rowsPerTeam), [&](PetscInt k) {
649         PetscInt i = t.league_rank() * rowsPerTeam + k; // i-th row in F
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));
940     mm->E_RowsPerTeam  = rowsPerTeam;
942     teamSize = vectorLength = rowsPerTeam = -1;
943     PetscCall(MatMergeGetLaunchParameters<DefaultExecutionSpace>(Fm, Fnz, -1, teamSize, vectorLength, rowsPerTeam));
946     mm->F_RowsPerTeam  = rowsPerTeam;
969   PetscInt rowsPerTeam  = mm->E_RowsPerTeam;
970   PetscInt workSets     = (irootloc.extent(0) + rowsPerTeam - 1) / rowsPerTeam;
975       Kokkos::parallel_for(Kokkos::TeamThreadRange(t, 0, rowsPerTeam), [&](PetscInt k) {
976         size_t r = t.league_rank() * rowsPerTeam + k; // r-th entry in irootloc[]
1018   PetscInt             rowsPerTeam  = mm->F_RowsPerTeam;
1019   PetscInt             workSets     = (Fm + rowsPerTeam - 1) / rowsPerTeam;
1026       Kokkos::parallel_for(Kokkos::TeamThreadRange(t, 0, rowsPerTeam), [&](PetscInt k) {
1027         PetscInt i = t.league_rank() * rowsPerTeam + k; // i-th row in F