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
64 lines
1.5 KiB
C++
64 lines
1.5 KiB
C++
#ifndef BENCHMARK_THREAD_MANAGER_H
|
|
#define BENCHMARK_THREAD_MANAGER_H
|
|
|
|
#include <atomic>
|
|
|
|
#include "benchmark/benchmark.h"
|
|
#include "mutex.h"
|
|
|
|
namespace benchmark {
|
|
namespace internal {
|
|
|
|
class ThreadManager {
|
|
public:
|
|
explicit ThreadManager(int num_threads)
|
|
: alive_threads_(num_threads), start_stop_barrier_(num_threads) {}
|
|
|
|
Mutex& GetBenchmarkMutex() const RETURN_CAPABILITY(benchmark_mutex_) {
|
|
return benchmark_mutex_;
|
|
}
|
|
|
|
bool StartStopBarrier() EXCLUDES(end_cond_mutex_) {
|
|
return start_stop_barrier_.wait();
|
|
}
|
|
|
|
void NotifyThreadComplete() EXCLUDES(end_cond_mutex_) {
|
|
start_stop_barrier_.removeThread();
|
|
if (--alive_threads_ == 0) {
|
|
MutexLock lock(end_cond_mutex_);
|
|
end_condition_.notify_all();
|
|
}
|
|
}
|
|
|
|
void WaitForAllThreads() EXCLUDES(end_cond_mutex_) {
|
|
MutexLock lock(end_cond_mutex_);
|
|
end_condition_.wait(lock.native_handle(),
|
|
[this]() { return alive_threads_ == 0; });
|
|
}
|
|
|
|
struct Result {
|
|
IterationCount iterations = 0;
|
|
double real_time_used = 0;
|
|
double cpu_time_used = 0;
|
|
double manual_time_used = 0;
|
|
int64_t complexity_n = 0;
|
|
std::string report_label_;
|
|
std::string error_message_;
|
|
bool has_error_ = false;
|
|
UserCounters counters;
|
|
};
|
|
GUARDED_BY(GetBenchmarkMutex()) Result results;
|
|
|
|
private:
|
|
mutable Mutex benchmark_mutex_;
|
|
std::atomic<int> alive_threads_;
|
|
Barrier start_stop_barrier_;
|
|
Mutex end_cond_mutex_;
|
|
Condition end_condition_;
|
|
};
|
|
|
|
} // namespace internal
|
|
} // namespace benchmark
|
|
|
|
#endif // BENCHMARK_THREAD_MANAGER_H
|