Skip testing for non-power-of-two team sizes

masterleinad · Sep 27, 2023 · 61e7b26 · 61e7b26
1 parent b8d4feb
commit 61e7b26
Showing 1 changed file with 2 additions and 9 deletions.
diff --git a/core/unit_test/TestTeamScan.hpp b/core/unit_test/TestTeamScan.hpp
@@ -185,16 +185,9 @@ struct TestTeamScanRetVal {
     a_r = view_2d_type("a_r", M, N);
     a_s = view_1d_type("a_s", M);
 
-    // Set team size explicitly to check whether non-power-of-two team sizes can
-    // be used.
-    if (ExecutionSpace().concurrency() > 10000)
-      Kokkos::parallel_for(policy_type(M, 127), *this);
-    else if (ExecutionSpace().concurrency() > 2)
-      Kokkos::parallel_for(policy_type(M, 3), *this);
-    else
-      Kokkos::parallel_for(policy_type(M, 1), *this);
-    Kokkos::fence();
+    Kokkos::parallel_for(policy_type(M, Kokkos::AUTO), *this);
 
+    Kokkos::fence();
     auto a_i  = Kokkos::create_mirror_view(a_d);
     auto a_o  = Kokkos::create_mirror_view(a_r);
     auto a_os = Kokkos::create_mirror_view(a_s);