add flops arg to benchmarker

shinra-dev · Jun 25, 2020 · 2d77fd8 · 2d77fd8
1 parent 4bb0df6
commit 2d77fd8
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/R/bench.r b/R/bench.r
@@ -204,7 +204,16 @@ benchR6 = R6::R6Class("bench",
 
       relative = round(private$elapsed / m, 2)
       tbl = cbind(private$elapsed, private$reps, avg, relative)
-      colnames(tbl) = private$get_colnames()
+      cn = private$get_colnames()
+
+      if (!is.null(private$flops))
+      {
+        flops = private$flops / avg / 1000^3
+        tbl = cbind(tbl, flops)
+        cn = c(cn, "gflops")
+      }
+
+      colnames(tbl) = cn
       rownames(tbl) = private$names
 
       tbl