Merge pull request #449 from ValeevGroup/evaleev/cleanup/gemm-examples

cleanup gemm examples

examples/CMakeLists.txt

@@ -30,7 +30,7 @@ endif()
 # Add Subdirectories
 add_subdirectory (cc)
 add_subdirectory (device)
-add_subdirectory (dgemm)
+add_subdirectory (gemm)
 add_subdirectory (demo)
 add_subdirectory (scalapack)
 add_subdirectory (fock)

examples/dgemm/CMakeLists.txt → examples/gemm/CMakeLists.txt

@@ -26,7 +26,7 @@
 # Create example executable
 
 foreach(_exec ta_blas ta_eigen ta_band ta_dense ta_sparse ta_dense_nonuniform
-              ta_dense_asymm ta_sparse_grow ta_dense_new_tile
+              ta_dense_asymm ta_sparse_grow
               ta_cc_abcd)
 
   # Add executable

examples/dgemm/README → examples/gemm/README

@@ -12,9 +12,9 @@ Applications usage:
 
   ta_band matrix_size block_size band_width [repetitions]
 
-  blas matrix_size [repetitions]
+  ta_blas matrix_size [repetitions]
 
-  eigen matrix_size [repetitions]
+  ta_eigen matrix_size [repetitions]
 
 Argument definitions:
 

examples/dgemm/ta_band.cpp → examples/gemm/ta_band.cpp

@@ -17,6 +17,7 @@
  *
  */
 
+#include <TiledArray/util/time.h>
 #include <tiledarray.h>
 #include <iostream>
 
@@ -104,38 +105,33 @@ int main(int argc, char** argv) {
       for (; j < j_end; ++j, ++ij) shape_tensor[ij] = 1.0;
     }
 
-    TiledArray::SparseShape<float> shape(shape_tensor, trange);
+    TiledArray::SparseShape<float> shape(
+        shape_tensor, trange, /* per_element_norms_already = */ true);
 
     // Construct and initialize arrays
     TiledArray::TSpArrayD a(world, trange, shape);
     TiledArray::TSpArrayD b(world, trange, shape);
-    TiledArray::TSpArrayD c(world, trange);
+    TiledArray::TSpArrayD c;
     a.fill(1.0);
     b.fill(1.0);
 
-    // Start clock
-    world.gop.fence();
-    const double wall_time_start = madness::wall_time();
-
     // Do matrix multiplication
+    world.gop.fence();
     for (int i = 0; i < repeat; ++i) {
-      c("m,n") = a("m,k") * b("k,n");
-      world.gop.fence();
+      TA_RECORD_DURATION(c("m,n") = a("m,k") * b("k,n"); world.gop.fence();)
       if (world.rank() == 0) std::cout << "Iteration " << i + 1 << "\n";
     }
 
-    // Stop clock
-    const double wall_time_stop = madness::wall_time();
-
     // Print results
-    const long flop = 2.0 * c("m,n").sum().get();
+    const auto gflops_per_call = 2.0 * c("m,n").sum().get() / 1.e9;
     if (world.rank() == 0) {
-      std::cout << "Average wall time = "
-                << (wall_time_stop - wall_time_start) / double(repeat)
-                << "\nAverage GFLOPS = "
-                << double(repeat) * double(flop) /
-                       (wall_time_stop - wall_time_start) / 1.0e9
-                << "\n";
+      auto durations = TiledArray::duration_statistics();
+      std::cout << "Average wall time   = " << durations.mean
+                << " s\nAverage GFLOPS      = "
+                << gflops_per_call * durations.mean_reciprocal
+                << "\nMedian wall time   = " << durations.median
+                << " s\nMedian GFLOPS      = "
+                << gflops_per_call / durations.median << "\n";
     }
 
   } catch (TiledArray::Exception& e) {

examples/dgemm/ta_blas.cpp → examples/gemm/ta_blas.cpp

@@ -17,13 +17,14 @@
  *
  */
 
+#include <TiledArray/util/time.h>
 #include <tiledarray.h>
 #include <iostream>
 
 int main(int argc, char** argv) {
   // Get command line arguments
   if (argc < 2) {
-    std::cout << "Usage: " << argv[0] << " matrix_size [repetitions]\n";
+    std::cout << "Usage: " << argv[0] << " matrix_size [repetitions = 5]\n";
     return 0;
   }
   const long matrix_size = atol(argv[1]);
@@ -66,31 +67,25 @@ int main(int argc, char** argv) {
   const integer m = matrix_size, n = matrix_size, k = matrix_size;
   const integer lda = matrix_size, ldb = matrix_size, ldc = matrix_size;
 
-  // Start clock
-  const double wall_time_start = madness::wall_time();
-
   // Do matrix multiplication
-  // Note: If TiledArray has not been configured with blas, this will be an
-  // eigen call.
   for (int i = 0; i < repeat; ++i) {
-    gemm(opa, opb, m, n, k, alpha, a, lda, b, ldb, beta, c, ldc);
+    TA_RECORD_DURATION(
+        gemm(opa, opb, m, n, k, alpha, a, lda, b, ldb, beta, c, ldc));
   }
-
-  // Stop clock
-  const double wall_time_stop = madness::wall_time();
+  auto durations = TiledArray::duration_statistics();
 
   // Cleanup memory
   free(a);
   free(b);
   free(c);
 
-  std::cout << "Average wall time = "
-            << (wall_time_stop - wall_time_start) / double(repeat)
-            << "\nAverage GFLOPS = "
-            << double(repeat) * 2.0 *
-                   double(matrix_size * matrix_size * matrix_size) /
-                   (wall_time_stop - wall_time_start) / 1.0e9
-            << "\n";
+  const auto gflops_per_call =
+      2.0 * double(matrix_size * matrix_size * matrix_size) / 1.0e9;
+  std::cout << "Average wall time = " << durations.mean << "\nAverage GFLOPS = "
+            << gflops_per_call * durations.mean_reciprocal
+            << "\nMedian wall time = " << durations.median
+            << "\nMedian GFLOPS = " << gflops_per_call / durations.median
+            << std::endl;
 
   return 0;
 }