Searched refs:teamSize (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 …PetscCall(MatMergeGetLaunchParameters<DefaultExecutionSpace>(Em, Enz, -1, teamSize, vectorLength, … in MatMPIAIJKokkosReduceBegin() 625 mm->E_TeamSize = teamSize; in MatMPIAIJKokkosReduceBegin() 640 PetscInt teamSize = mm->E_TeamSize; in MatMPIAIJKokkosReduceBegin() local 647 …Kokkos::TeamPolicy<>(PetscGetKokkosExecutionSpace(), workSets, teamSize, vectorLength), KOKKOS_LAM… in MatMPIAIJKokkosReduceBegin() 934 PetscInt teamSize, vectorLength, rowsPerTeam; in MatMPIAIJKokkosBcastBegin() local 936 teamSize = vectorLength = rowsPerTeam = -1; in MatMPIAIJKokkosBcastBegin() 937 …faultExecutionSpace>(mm->irootloc.extent(0), mm->rootBuf.extent(0), -1, teamSize, vectorLength, ro… in MatMPIAIJKokkosBcastBegin() 938 mm->E_TeamSize = teamSize; in MatMPIAIJKokkosBcastBegin() [all …]
|