567bfb3779
The * flag to printf() wants an int instead of size_t, and these are distinct types on 64-bit. To accomodate this, make the name column width helpers return int. In theory this truncates things, but in practice this only matters if you have a benchmark with more than INT_MAX characters in its name (in which case you have bigger problems). Change-Id: I3338948c25a3a8d84f1ead2f5b457c05da8a01cf Signed-off-by: Greg Hackmann <ghackmann@google.com>
154 lines
3.8 KiB
C++
154 lines
3.8 KiB
C++
/*
|
|
* Copyright (C) 2012 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include <inttypes.h>
|
|
#include <regex.h>
|
|
#include <stdio.h>
|
|
#include <stdint.h>
|
|
#include <stdlib.h>
|
|
#include <time.h>
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include <utils/stringprintf.h>
|
|
|
|
#include <benchmark/Benchmark.h>
|
|
|
|
#include "utils.h"
|
|
|
|
namespace testing {
|
|
|
|
static uint64_t NanoTime() {
|
|
struct timespec t;
|
|
t.tv_sec = t.tv_nsec = 0;
|
|
clock_gettime(CLOCK_MONOTONIC, &t);
|
|
return static_cast<uint64_t>(t.tv_sec) * 1000000000LL + t.tv_nsec;
|
|
}
|
|
|
|
std::vector<Benchmark*>& Benchmark::List() {
|
|
static std::vector<Benchmark*> list;
|
|
return list;
|
|
}
|
|
|
|
int Benchmark::MaxNameColumnWidth() {
|
|
int max = 20;
|
|
for (auto& benchmark : List()) {
|
|
max = std::max(max, benchmark->NameColumnWidth());
|
|
}
|
|
return max;
|
|
}
|
|
|
|
bool Benchmark::RunAll(std::vector<regex_t*>& regs) {
|
|
bool ran_benchmark = false;
|
|
for (auto& benchmark : List()) {
|
|
if (benchmark->ShouldRun(regs)) {
|
|
if (!ran_benchmark) {
|
|
printf("%-*s %10s %10s\n", MaxNameColumnWidth(), "", "iterations", "ns/op");
|
|
ran_benchmark = true;
|
|
}
|
|
benchmark->RunAll();
|
|
}
|
|
}
|
|
return ran_benchmark;
|
|
}
|
|
|
|
bool Benchmark::ShouldRun(std::vector<regex_t*>& regs) {
|
|
if (regs.empty()) {
|
|
return true;
|
|
}
|
|
|
|
for (const auto& re : regs) {
|
|
if (regexec(re, Name().c_str(), 0, NULL, 0) != REG_NOMATCH) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
void Benchmark::StopBenchmarkTiming() {
|
|
if (start_time_ns_ != 0) {
|
|
total_time_ns_ += NanoTime() - start_time_ns_;
|
|
}
|
|
start_time_ns_ = 0;
|
|
}
|
|
|
|
void Benchmark::StartBenchmarkTiming() {
|
|
if (start_time_ns_ == 0) {
|
|
start_time_ns_ = NanoTime();
|
|
}
|
|
}
|
|
|
|
std::string BenchmarkWithoutArg::GetNameStr(void*) {
|
|
return Name();
|
|
}
|
|
|
|
template <>
|
|
std::string BenchmarkWithArg<int>::GetNameStr(int arg) {
|
|
return Name() + "/" + PrettyInt(arg, 2);
|
|
}
|
|
|
|
template <>
|
|
std::string BenchmarkWithArg<double>::GetNameStr(double arg) {
|
|
return Name() + "/" + android::StringPrintf("%0.6f", arg);
|
|
}
|
|
|
|
template<typename T>
|
|
void BenchmarkT<T>::RunWithArg(T arg) {
|
|
int new_iterations = 1;
|
|
int iterations;
|
|
while (new_iterations < 1e8) {
|
|
bytes_processed_ = 0;
|
|
total_time_ns_ = 0;
|
|
start_time_ns_ = 0;
|
|
|
|
iterations = new_iterations;
|
|
RunIterations(iterations, arg);
|
|
if (total_time_ns_ >= 1e9) {
|
|
break;
|
|
}
|
|
|
|
if (total_time_ns_/iterations == 0) {
|
|
new_iterations = 1e9;
|
|
} else {
|
|
new_iterations = 1e9/ (total_time_ns_/iterations);
|
|
}
|
|
new_iterations = std::max(iterations + 1,
|
|
std::min(new_iterations + new_iterations/2, 100*iterations));
|
|
|
|
new_iterations = Round(new_iterations);
|
|
}
|
|
|
|
printf("%-*s %10s %10" PRId64, MaxNameColumnWidth(), GetNameStr(arg).c_str(),
|
|
PrettyInt(iterations, 10).c_str(), total_time_ns_/iterations);
|
|
|
|
if (total_time_ns_ > 0 && bytes_processed_ > 0) {
|
|
double gib_processed = static_cast<double>(bytes_processed_)/1e9;
|
|
double seconds = static_cast<double>(total_time_ns_)/1e9;
|
|
printf(" %8.3f GiB/s", gib_processed/seconds);
|
|
}
|
|
printf("\n");
|
|
fflush(stdout);
|
|
}
|
|
|
|
template class BenchmarkT<int>;
|
|
template class BenchmarkT<double>;
|
|
template class BenchmarkT<void*>;
|
|
|
|
template class BenchmarkWithArg<int>;
|
|
template class BenchmarkWithArg<double>;
|
|
|
|
} // namespace testing
|