Lines Matching refs:vectorLength

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()
626 mm->E_VectorLength = vectorLength; in MatMPIAIJKokkosReduceBegin()
641 PetscInt vectorLength = mm->E_VectorLength; 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 …ionSpace>(mm->irootloc.extent(0), mm->rootBuf.extent(0), -1, teamSize, vectorLength, rowsPerTeam)); in MatMPIAIJKokkosBcastBegin()
939 mm->E_VectorLength = vectorLength; in MatMPIAIJKokkosBcastBegin()
942 teamSize = vectorLength = rowsPerTeam = -1; in MatMPIAIJKokkosBcastBegin()
943 …MergeGetLaunchParameters<DefaultExecutionSpace>(Fm, Fnz, -1, teamSize, vectorLength, rowsPerTeam)); in MatMPIAIJKokkosBcastBegin()
945 mm->F_VectorLength = vectorLength; in MatMPIAIJKokkosBcastBegin()
968 PetscInt vectorLength = mm->E_VectorLength; in MatMPIAIJKokkosBcastBegin() local
974 …Kokkos::TeamPolicy<>(PetscGetKokkosExecutionSpace(), workSets, teamSize, vectorLength), KOKKOS_LAM… in MatMPIAIJKokkosBcastBegin()
1017 PetscInt vectorLength = mm->F_VectorLength; in MatMPIAIJKokkosBcastEnd() local
1025 …Kokkos::TeamPolicy<>(PetscGetKokkosExecutionSpace(), workSets, teamSize, vectorLength), KOKKOS_LAM… in MatMPIAIJKokkosBcastEnd()