gemm perf_test: print matrix sizes (kokkos#2362)

lucbv · Oct 9, 2024 · 773e47a · 773e47a
1 parent 6042bba
commit 773e47a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/perf_test/blas/blas3/KokkosBlas3_gemm_standalone_perf_test.cpp b/perf_test/blas/blas3/KokkosBlas3_gemm_standalone_perf_test.cpp
@@ -103,7 +103,8 @@ template <typename ExecSpace>
 void run(int m, int n, int k, int repeat) {
   using LL = Kokkos::LayoutLeft;
   using LR = Kokkos::LayoutRight;
-  std::cout << "** Running GEMM experiments (" << ExecSpace::name() << ") **\n";
+  std::cout << "** Running GEMM experiments (" << ExecSpace::name() << " m=" << m << " n=" << n << " k=" << k
+            << ") **\n";
   std::cout << "Running: A LayoutLeft, B LayoutLeft  : ";
   runImpl<ExecSpace, LL, LL>(m, n, k, repeat);
   std::cout << "Running: A LayoutLeft, B LayoutRight : ";