KokkosBlas_team_dot_perf_test: refactor w/ functor

kokkos · Sep 28, 2021 · 06f0ddb · 06f0ddb
1 parent d62332e
commit 06f0ddb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/perf_test/blas/blas1/KokkosBlas_team_dot_perf_test.cpp b/perf_test/blas/blas1/KokkosBlas_team_dot_perf_test.cpp
@@ -149,8 +149,8 @@ void run(int m, int repeat) {
   std::cout << "Each test input vector has a length of " << m << std::endl;
 
   // Warm up run of dot:
-  teamDotFunctor<Kokkos::View<Scalar*>, ExecSpace> teamDotFunctorWarmUpInstance(
-      x, y);
+  teamDotFunctor<Kokkos::View<Scalar*, MemSpace>, ExecSpace>
+      teamDotFunctorWarmUpInstance(x, y);
 
   Kokkos::parallel_for("TeamDotUsage -- Warm Up Run", policy(1, Kokkos::AUTO),
                        teamDotFunctorWarmUpInstance);
@@ -160,7 +160,7 @@ void run(int m, int repeat) {
 
   // Live test of dot:
 
-  teamDotFunctor<Kokkos::View<Scalar*>, ExecSpace>
+  teamDotFunctor<Kokkos::View<Scalar*, MemSpace>, ExecSpace>
       teamDotFunctorLiveTestInstance(x, y);
   Kokkos::parallel_for("TeamDotUsage -- Live Test", policy(1, Kokkos::AUTO),
                        teamDotFunctorLiveTestInstance);

diff --git a/perf_test/blas/blas1/KokkosBlas_team_dot_perf_test.hpp b/perf_test/blas/blas1/KokkosBlas_team_dot_perf_test.hpp
@@ -20,7 +20,7 @@ struct testData_rps_team_dot {
   using MemSpace = typename ExecSpace::memory_space;
   using Device   = Kokkos::Device<ExecSpace, MemSpace>;
 
-  using policy      = Kokkos::TeamPolicy<>;
+  using policy      = Kokkos::TeamPolicy<ExecSpace>;
   using member_type = typename policy::member_type;
   int numberOfTeams = 1;  // This will be passed to the TeamPolicy
   // Test Matrices x and y, View declaration