c_bench: report time in ms

Author: siboehm

benchmarks/c_bench/c_bench.cpp

@@ -5,7 +5,8 @@
 #include <cstdlib>
 #include <iostream>
 
-static void bm_lleaves(benchmark::State& state) {
+static void bm_lleaves(benchmark::State &state)
+{
   char *model_name = std::getenv("LLEAVES_BENCHMARK_MODEL");
 
   std::ostringstream model_stream;
@@ -17,11 +18,12 @@ static void bm_lleaves(benchmark::State& state) {
   ulong n_preds = arr.shape[0];
   auto *out = (double *)(malloc(n_preds * sizeof(double)));
 
-  for (auto _ : state){
-      // predict over the whole input array
-      forest_root(loaded_data, out, (int)0, (int)n_preds);
+  for (auto _ : state)
+  {
+    // predict over the whole input array
+    forest_root(loaded_data, out, (int)0, (int)n_preds);
   }
 }
 
-BENCHMARK(bm_lleaves);
+BENCHMARK(bm_lleaves)->Unit(benchmark::kMillisecond);
 BENCHMARK_MAIN();

benchmarks/c_bench/c_bench.h

@@ -5,8 +5,9 @@
 #ifndef C_BENCH_LLVM_H
 #define C_BENCH_LLVM_H
 
-extern "C"{
-void forest_root(double *, double *, int, int);
+extern "C"
+{
+    void forest_root(double *, double *, int, int);
 }
 
-#endif //C_BENCH_LLVM_H
+#endif // C_BENCH_LLVM_H

benchmarks/data/setup_data.sh

@@ -1,6 +1,6 @@
 #!/usr/bin/env bash
 
-set -euo pipefail
+set -euox pipefail
 
 pushd benchmarks/data/
 wget -q https://f003.backblazeb2.com/file/lleaves-benchmark/benchmark_data.zip