Searched refs:rowsPerTeam (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 621 PetscInt teamSize, vectorLength, rowsPerTeam; in MatMPIAIJKokkosReduceBegin() local
623 teamSize = vectorLength = rowsPerTeam = -1; in MatMPIAIJKokkosReduceBegin()
624 …MergeGetLaunchParameters<DefaultExecutionSpace>(Em, Enz, -1, teamSize, vectorLength, rowsPerTeam)); in MatMPIAIJKokkosReduceBegin()
627 mm->E_RowsPerTeam = rowsPerTeam; in MatMPIAIJKokkosReduceBegin()
642 PetscInt rowsPerTeam = mm->E_RowsPerTeam; in MatMPIAIJKokkosReduceBegin() local
643 PetscInt workSets = (Em + rowsPerTeam - 1) / rowsPerTeam; in MatMPIAIJKokkosReduceBegin()
648 Kokkos::parallel_for(Kokkos::TeamThreadRange(t, 0, rowsPerTeam), [&](PetscInt k) { in MatMPIAIJKokkosReduceBegin()
649 PetscInt i = t.league_rank() * rowsPerTeam + k; // i-th row in F in MatMPIAIJKokkosReduceBegin()
934 PetscInt teamSize, vectorLength, rowsPerTeam; in MatMPIAIJKokkosBcastBegin() local
936 teamSize = vectorLength = rowsPerTeam = -1; in MatMPIAIJKokkosBcastBegin()
[all …]
|