add gather benchmark to CMake

masterleinad · Mar 4, 2024 · 932466f · 932466f
1 parent 5c9a4aa
commit 932466f
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 10 deletions.
diff --git a/benchmarks/CMakeLists.txt b/benchmarks/CMakeLists.txt
@@ -1,4 +1,5 @@
 KOKKOS_ADD_BENCHMARK_DIRECTORIES(bytes_and_flops)
+KOKKOS_ADD_BENCHMARK_DIRECTORIES(gather)
 KOKKOS_ADD_BENCHMARK_DIRECTORIES(gups)
 KOKKOS_ADD_BENCHMARK_DIRECTORIES(launch_latency)
 KOKKOS_ADD_BENCHMARK_DIRECTORIES(stream)
diff --git a/benchmarks/gather/CMakeLists.txt b/benchmarks/gather/CMakeLists.txt
@@ -0,0 +1,4 @@
+KOKKOS_ADD_EXECUTABLE(
+  gather
+  SOURCES main.cpp
+)
diff --git a/benchmarks/gather/gather.hpp b/benchmarks/gather/gather.hpp
@@ -20,28 +20,28 @@ struct RunGather {
 };
 
 #define UNROLL 1
-#include <gather_unroll.hpp>
+#include "gather_unroll.hpp"
 #undef UNROLL
 #define UNROLL 2
-#include <gather_unroll.hpp>
+#include "gather_unroll.hpp"
 #undef UNROLL
 #define UNROLL 3
-#include <gather_unroll.hpp>
+#include "gather_unroll.hpp"
 #undef UNROLL
 #define UNROLL 4
-#include <gather_unroll.hpp>
+#include "gather_unroll.hpp"
 #undef UNROLL
 #define UNROLL 5
-#include <gather_unroll.hpp>
+#include "gather_unroll.hpp"
 #undef UNROLL
 #define UNROLL 6
-#include <gather_unroll.hpp>
+#include "gather_unroll.hpp"
 #undef UNROLL
 #define UNROLL 7
-#include <gather_unroll.hpp>
+#include "gather_unroll.hpp"
 #undef UNROLL
 #define UNROLL 8
-#include <gather_unroll.hpp>
+#include "gather_unroll.hpp"
 #undef UNROLL
 
 template <class Scalar>

diff --git a/benchmarks/gather/gather_unroll.hpp b/benchmarks/gather/gather_unroll.hpp
@@ -138,7 +138,7 @@ struct RunGather<Scalar, UNROLL> {
     printf(
         "SNKDRUF: %i %i %i %i %i %i %i Time: %lfs Bandwidth: %lfGiB/s GFlop/s: "
         "%lf GGather/s: %lf\n",
-        sizeof(Scalar) / 4, N, K, D, R, UNROLL, F, seconds,
+        static_cast<int>(sizeof(Scalar) / 4), N, K, D, R, UNROLL, F, seconds,
         1.0 * bytes / seconds / 1024 / 1024 / 1024, 1.e-9 * flops / seconds,
         1.e-9 * gather_ops / seconds);
   }

diff --git a/benchmarks/gather/main.cpp b/benchmarks/gather/main.cpp
@@ -16,7 +16,7 @@
 
 #include <Kokkos_Core.hpp>
 #include <Kokkos_Timer.hpp>
-#include <gather.hpp>
+#include "gather.hpp"
 #include <cstdlib>
 
 int main(int argc, char* argv[]) {