From beacb6f9a0bdd4bb85a9978005d0a3b7f12f2c42 Mon Sep 17 00:00:00 2001 From: Philipp Grete Date: Wed, 18 Dec 2024 23:43:04 +0100 Subject: [PATCH] Team size 256 --- src/kokkos_abstraction.hpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/kokkos_abstraction.hpp b/src/kokkos_abstraction.hpp index f2fa929a8f16..855b6ca4622e 100644 --- a/src/kokkos_abstraction.hpp +++ b/src/kokkos_abstraction.hpp @@ -724,7 +724,7 @@ inline void par_for_outer(OuterLoopPatternTeams, const std::string &name, const Function &function) { const int Nk = ku + 1 - kl; - team_policy policy(exec_space, Nk, 192); + team_policy policy(exec_space, Nk, 256); Kokkos::parallel_for( name, @@ -745,7 +745,7 @@ inline void par_for_outer(OuterLoopPatternTeams, const std::string &name, const int Nj = ju + 1 - jl; const int NkNj = Nk * Nj; - team_policy policy(exec_space, NkNj, 192); + team_policy policy(exec_space, NkNj, 256); Kokkos::parallel_for( name, @@ -770,7 +770,7 @@ inline void par_for_outer(OuterLoopPatternTeams, const std::string &name, const int NkNj = Nk * Nj; const int NnNkNj = Nn * Nk * Nj; - team_policy policy(exec_space, NnNkNj, 192); + team_policy policy(exec_space, NnNkNj, 256); Kokkos::parallel_for( name,