Searched refs:workSets (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 643 PetscInt workSets = (Em + rowsPerTeam - 1) / rowsPerTeam; in MatMPIAIJKokkosReduceBegin() local
647 …Kokkos::TeamPolicy<>(PetscGetKokkosExecutionSpace(), workSets, teamSize, vectorLength), KOKKOS_LAM… in MatMPIAIJKokkosReduceBegin()
970 PetscInt workSets = (irootloc.extent(0) + rowsPerTeam - 1) / rowsPerTeam; in MatMPIAIJKokkosBcastBegin() local
974 …Kokkos::TeamPolicy<>(PetscGetKokkosExecutionSpace(), workSets, teamSize, vectorLength), KOKKOS_LAM… in MatMPIAIJKokkosBcastBegin()
1019 PetscInt workSets = (Fm + rowsPerTeam - 1) / rowsPerTeam; in MatMPIAIJKokkosBcastEnd() local
1025 …Kokkos::TeamPolicy<>(PetscGetKokkosExecutionSpace(), workSets, teamSize, vectorLength), KOKKOS_LAM… in MatMPIAIJKokkosBcastEnd()
|