Following TACC hackathon set teamsize to 192

parthenon-hpc-lab · Dec 18, 2024 · ffe3de1 · ffe3de1
1 parent 6461074
commit ffe3de1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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, Kokkos::AUTO);
+  team_policy policy(exec_space, Nk, 192);
 
   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, Kokkos::AUTO);
+  team_policy policy(exec_space, NkNj, 192);
 
   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, Kokkos::AUTO);
+  team_policy policy(exec_space, NnNkNj, 192);
 
   Kokkos::parallel_for(
       name,