perf_tests: Initialize Kokkos

kokkos · Dec 18, 2023 · 907aa9d · 907aa9d
1 parent 74d7d8b
commit 907aa9d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/perf_tests/test_main.cpp b/perf_tests/test_main.cpp
@@ -1,5 +1,6 @@
 #include <mpi.h>
 #include <benchmark/benchmark.h>
+#include <Kokkos_Core.hpp>
 
 // This reporter does nothing.
 // We can use it to disable output from all but the root process
@@ -20,6 +21,8 @@ int main(int argc, char **argv) {
   int rank;
   MPI_Comm_rank(MPI_COMM_WORLD, &rank);
 
+  Kokkos::initialize();
+
   ::benchmark::Initialize(&argc, argv);
 
   if(rank == 0)
@@ -32,6 +35,7 @@ int main(int argc, char **argv) {
     ::benchmark::RunSpecifiedBenchmarks(&null);
   }
 
+  Kokkos::finalize();
   MPI_Finalize();
   return 0;
 }