Use Policy for all parallel_for tests

gojakuch · Apr 9, 2024 · ca62619 · ca62619
1 parent 17fc64b
commit ca62619
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/unittests/Kokkos/ParallelFor.cpp b/unittests/Kokkos/ParallelFor.cpp
@@ -10,8 +10,9 @@ TEST(ParallelFor, HelloWorldLambdaLoopForward) {
     // const double tau = 1e-6; // tolerance
 
     std::function<double(double)> _f = [](double x) {
+        using Policy = Kokkos::RangePolicy<Kokkos::DefaultHostExecutionSpace>;
         double res[5] = {0};
-        Kokkos::parallel_for("HelloWorld-forward", 5,
+        Kokkos::parallel_for("HelloWorld-forward", Policy(0, 5),
         [&res, x](const int i) {
             res[i] = x*x;
         });
@@ -34,8 +35,9 @@ TEST(ParallelFor, HelloWorldLambdaLoopReverse) {
     // const double tau = 1e-6; // tolerance
 
     std::function<double(double)> _f = [](double x) {
+        using Policy = Kokkos::RangePolicy<Kokkos::DefaultHostExecutionSpace>;
         double res[5] = {0};
-        Kokkos::parallel_for("HelloWorld-reverse", 5,
+        Kokkos::parallel_for("HelloWorld-reverse", Policy(0, 5),
         [&res, x](const int i) {
             res[i] = x*x;
         });