Codemod: use #include angle brackets in folly and thrift

[folly.git] / folly / gen / test / Bench.h

diff --git a/folly/gen/test/Bench.h b/folly/gen/test/Bench.h
index be91357faabb9d459e00e6e392691d494399487b..746553bddb0a9ec36218e3674514e2196d4ceb9b 100644 (file)
--- a/folly/gen/test/Bench.h
+++ b/folly/gen/test/Bench.h
@@ -17,15 +17,17 @@
 #ifndef FOLLY_GEN_BENCH_H_
 #define FOLLY_GEN_BENCH_H_
 
-#include "folly/Benchmark.h"
+#include <folly/Benchmark.h>
 
 #define BENCH_GEN_IMPL(gen, prefix)                         \
 static bool FB_ANONYMOUS_VARIABLE(benchGen) = (             \
   ::folly::addBenchmark(__FILE__, prefix FB_STRINGIZE(gen), \
     [](unsigned iters){                                     \
+      const unsigned num = iters;                           \
       while (iters--) {                                     \
         folly::doNotOptimizeAway(gen);                      \
       }                                                     \
+      return num;                                           \
     }), true)
 #define BENCH_GEN(gen) BENCH_GEN_IMPL(gen, "")
 #define BENCH_GEN_REL(gen) BENCH_GEN_IMPL(gen, "%")