a290770fa8
Pulled the latest (b000672), which adds an option to disable -Werror, which caused issues previously (see for instance https://reviews.llvm.org/D112012) Applied https://github.com/google/benchmark/pull/1305 to unblock, as this issue is causing downstream buildbot failures. The only change is in llvm/CMakeLists.txt, where we set BENCHMARK_ENABLE_WERROR to have the same value as LLVM_ENABLE_WERROR. This patch also introduces a convenience script to pull benchmark from upstream. Differential Revision: https://reviews.llvm.org/D115684
52 lines
1.2 KiB
C++
52 lines
1.2 KiB
C++
|
|
#include <cassert>
|
|
#include <memory>
|
|
|
|
#include "benchmark/benchmark.h"
|
|
|
|
#define FIXTURE_BECHMARK_NAME MyFixture
|
|
|
|
class FIXTURE_BECHMARK_NAME : public ::benchmark::Fixture {
|
|
public:
|
|
void SetUp(const ::benchmark::State& state) BENCHMARK_OVERRIDE {
|
|
if (state.thread_index() == 0) {
|
|
assert(data.get() == nullptr);
|
|
data.reset(new int(42));
|
|
}
|
|
}
|
|
|
|
void TearDown(const ::benchmark::State& state) BENCHMARK_OVERRIDE {
|
|
if (state.thread_index() == 0) {
|
|
assert(data.get() != nullptr);
|
|
data.reset();
|
|
}
|
|
}
|
|
|
|
~FIXTURE_BECHMARK_NAME() { assert(data == nullptr); }
|
|
|
|
std::unique_ptr<int> data;
|
|
};
|
|
|
|
BENCHMARK_F(FIXTURE_BECHMARK_NAME, Foo)(benchmark::State& st) {
|
|
assert(data.get() != nullptr);
|
|
assert(*data == 42);
|
|
for (auto _ : st) {
|
|
}
|
|
}
|
|
|
|
BENCHMARK_DEFINE_F(FIXTURE_BECHMARK_NAME, Bar)(benchmark::State& st) {
|
|
if (st.thread_index() == 0) {
|
|
assert(data.get() != nullptr);
|
|
assert(*data == 42);
|
|
}
|
|
for (auto _ : st) {
|
|
assert(data.get() != nullptr);
|
|
assert(*data == 42);
|
|
}
|
|
st.SetItemsProcessed(st.range(0));
|
|
}
|
|
BENCHMARK_REGISTER_F(FIXTURE_BECHMARK_NAME, Bar)->Arg(42);
|
|
BENCHMARK_REGISTER_F(FIXTURE_BECHMARK_NAME, Bar)->Arg(42)->ThreadPerCpu();
|
|
|
|
BENCHMARK_MAIN();
|