77 lines
2.7 KiB
C++
77 lines
2.7 KiB
C++
// Copyright (C) 2022 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 <getopt.h>
|
|
#include <unistd.h>
|
|
|
|
#include <cstring>
|
|
#include <filesystem>
|
|
#include <string>
|
|
|
|
#include <benchmark/benchmark.h>
|
|
#include <ditto/arg_parser.h>
|
|
#include <ditto/logger.h>
|
|
#include <ditto/parser.h>
|
|
|
|
template <class... Args>
|
|
void BM_DittoBench(benchmark::State& state, Args&&... args) {
|
|
for (auto _ : state) {
|
|
auto args_tuple = std::make_tuple(std::move(args)...);
|
|
|
|
std::error_code errorCode;
|
|
auto myPath = std::filesystem::path("/proc/self/exe");
|
|
auto executable_path = std::filesystem::canonical(myPath, errorCode);
|
|
auto project_path = executable_path.parent_path();
|
|
std::string ditto_path = project_path.string() + "/" + std::get<0>(args_tuple);
|
|
|
|
// LOGW("Opening ditto file: " + ditto_path);
|
|
// TODO error code?
|
|
|
|
dittosuite::CmdArguments arguments = {.results_output = dittosuite::ResultsOutput::kNull,
|
|
.file_path = ditto_path};
|
|
|
|
dittosuite::Parser::GetParser().Parse(arguments.file_path, arguments.parameters);
|
|
|
|
auto init = dittosuite::Parser::GetParser().GetInit();
|
|
if (init) {
|
|
init->SetUp();
|
|
init->Run();
|
|
init->TearDown();
|
|
}
|
|
|
|
auto main = dittosuite::Parser::GetParser().GetMain();
|
|
main->SetUp();
|
|
main->Run();
|
|
main->TearDown();
|
|
|
|
auto result = main->CollectResults("");
|
|
double duration_ns = result->GetSamples("duration")[0];
|
|
state.SetIterationTime(duration_ns / 1000 / 1000 / 1000);
|
|
|
|
auto clean_up = dittosuite::Parser::GetParser().GetCleanUp();
|
|
if (clean_up) {
|
|
clean_up->SetUp();
|
|
clean_up->Run();
|
|
clean_up->TearDown();
|
|
}
|
|
}
|
|
}
|
|
|
|
BENCHMARK_CAPTURE(BM_DittoBench, read_random_native, "example/android/read_random_native.ditto")->UseManualTime();
|
|
BENCHMARK_CAPTURE(BM_DittoBench, read_random_fuse, "example/android/read_random_fuse.ditto")->UseManualTime();
|
|
BENCHMARK_CAPTURE(BM_DittoBench, write_random_native, "example/android/write_random_native.ditto")->UseManualTime();
|
|
BENCHMARK_CAPTURE(BM_DittoBench, write_random_fuse, "example/android/write_random_fuse.ditto")->UseManualTime();
|
|
|
|
BENCHMARK_MAIN();
|