fix clbas

undertherain · Oct 15, 2020 · 825f293 · 825f293
1 parent 87ef0e3
commit 825f293
Showing 1 changed file with 28 additions and 21 deletions.
diff --git a/benchmarker/modules/problems/gemm/cblas/main.cpp b/benchmarker/modules/problems/gemm/cblas/main.cpp
@@ -10,39 +10,46 @@ int main(int argc, char * argv[]) {
     size_t m, n, k;
     float *A, *B, *C;
     double dtime;
-    std::string precision;
-    parse_args(argc, argv, precision, m, k, n);
+    Options options = parse_args(argc, argv);
+    // parse_args(argc, argv, precision, m, k, n);
+    m = options.cnt_rows_A_rows_C;
+    n = options.cnt_cols_A_rows_B;
+    k = options.cnt_cols_B_cols_C;
     get_matrices<float>(m, k, n, A, B, C);
     const float alpha = 1;
     const float beta = 0;
     const size_t lda=m; // k for row major;
     const size_t ldb=k; //n; 
     const size_t ldc=m; //n;
     auto start = high_resolution_clock::now(); 
-    if (precision == "FP32")
-        cblas_sgemm(CblasColMajor,
-                    CblasNoTrans,
-                    CblasNoTrans,
-                    m,
-                    n,
-                    k,
-                    alpha,
-                    A, lda,
-                    B, ldb,
-                    beta,
-                    C, ldc);
-    else
-	{
-        // TODO  (Alex): implement FP16
-        // ugly throw here to make sure benchmarker chrashes alright
-        fprintf(stderr, "not implemented yet");
-	    throw "madamada";
-	}
+    for (size_t i=0; i<options.nb_epoch; i++)
+    {
+        if (options.precision == "FP32")
+            cblas_sgemm(CblasColMajor,
+                        CblasNoTrans,
+                        CblasNoTrans,
+                        m,
+                        n,
+                        k,
+                        alpha,
+                        A, lda,
+                        B, ldb,
+                        beta,
+                        C, ldc);
+        else
+        {
+            // TODO  (Alex): implement FP16
+            // ugly throw here to make sure benchmarker chrashes alright
+            fprintf(stderr, "not implemented yet");
+            throw "madamada";
+        }
+    }
     std::cerr << "MNK " << m << " " << n << " " << k << std::endl;
     auto stop = high_resolution_clock::now();
     std::chrono::duration<double> seconds = (stop - start); 
     dtime = seconds.count();
     double gflop = (2.0 * m * n * k) / (1024 * 1024 * 1024);
+    gflop *= static_cast<double>(options.nb_epoch);
     double gflops = gflop / dtime;
     printf("%f\n", dtime);
     fprintf(stderr, "gflops: \t%f\n", gflop);