mirror of
https://github.com/cirosantilli/linux-kernel-module-cheat.git
synced 2026-01-23 02:05:57 +01:00
175b0e3d50
One data structure per executable, and gem5 instrumented versions in separate executables. Create the allowed_emulators path_properties property. TODO: wire it up to benchmark result generation and remove the old version.
116 lines
2.7 KiB
C++
116 lines
2.7 KiB
C++
// https://cirosantilli.com/linux-kernel-module-cheat#bst-vs-heap-vs-hashmap
|
|
|
|
#include <algorithm>
|
|
#include <cassert>
|
|
#include <chrono>
|
|
#include <iostream>
|
|
#if LKMC_ALGORITHM_SET_STD_PRIORITY_QUEUE
|
|
#include <queue>
|
|
#endif
|
|
#include <random>
|
|
#if LKMC_ALGORITHM_SET_STD_SET
|
|
#include <set>
|
|
#endif
|
|
#if LKMC_ALGORITHM_SET_STD_UNORDERED_SET
|
|
#include <unordered_set>
|
|
#endif
|
|
|
|
#include <lkmc/m5ops.h>
|
|
|
|
int main(int argc, char **argv) {
|
|
typedef uint64_t T;
|
|
#if LKMC_ALGORITHM_SET_STD_PRIORITY_QUEUE
|
|
std::priority_queue<T> set;
|
|
#endif
|
|
#if LKMC_ALGORITHM_SET_STD_SET
|
|
std::set<T> set;
|
|
#endif
|
|
#if LKMC_ALGORITHM_SET_STD_UNORDERED_SET
|
|
std::unordered_set<T> set;
|
|
#endif
|
|
std::vector<T> randoms;
|
|
size_t i, j = 0, n, granule, base;
|
|
unsigned int seed;
|
|
#ifndef LKMC_M5OPS_ENABLE
|
|
std::vector<std::chrono::nanoseconds::rep> dts;
|
|
std::vector<decltype(base)> bases;
|
|
#endif
|
|
|
|
// CLI arguments.
|
|
if (argc > 1) {
|
|
n = std::stoi(argv[1]);
|
|
} else {
|
|
n = 10;
|
|
}
|
|
if (argc > 2) {
|
|
granule = std::stoi(argv[2]);
|
|
} else {
|
|
granule = 1;
|
|
}
|
|
if (argc > 3) {
|
|
seed = std::stoi(argv[3]);
|
|
} else {
|
|
seed = std::random_device()();
|
|
}
|
|
|
|
// Action.
|
|
for (i = 0; i < n; ++i) {
|
|
randoms.push_back(i);
|
|
}
|
|
std::shuffle(randoms.begin(), randoms.end(), std::mt19937(seed));
|
|
for (i = 0; i < n / granule; ++i) {
|
|
#ifndef LKMC_M5OPS_ENABLE
|
|
using clk = std::chrono::high_resolution_clock;
|
|
decltype(clk::now()) start, end;
|
|
#endif
|
|
base = i * granule;
|
|
#ifdef LKMC_M5OPS_ENABLE
|
|
LKMC_M5OPS_RESETSTATS;
|
|
#else
|
|
start = clk::now();
|
|
for (j = 0; j < granule; ++j) {
|
|
#endif
|
|
#if LKMC_ALGORITHM_SET_STD_PRIORITY_QUEUE
|
|
set.emplace(randoms[base + j]);
|
|
#else
|
|
set.insert(randoms[base + j]);
|
|
#endif
|
|
#ifdef LKMC_M5OPS_ENABLE
|
|
LKMC_M5OPS_DUMPSTATS;
|
|
#else
|
|
}
|
|
end = clk::now();
|
|
auto dt = (end - start) / granule;
|
|
bases.push_back(base);
|
|
dts.push_back(std::chrono::duration_cast<std::chrono::nanoseconds>(dt).count());
|
|
#endif
|
|
}
|
|
|
|
// Report results.
|
|
#ifndef LKMC_M5OPS_ENABLE
|
|
// Output.
|
|
std::cout << "times" << std::endl;
|
|
auto bases_it = bases.begin();
|
|
auto dts_it = dts.begin();
|
|
while(bases_it != bases.end()) {
|
|
const auto& base = *bases_it;
|
|
const auto& dt = *dts_it;
|
|
std::cout << base << " " << dt << std::endl;
|
|
bases_it++;
|
|
dts_it++;
|
|
}
|
|
std::cout << std::endl;
|
|
std::cout << "output" << std::endl;
|
|
#if LKMC_ALGORITHM_SET_STD_PRIORITY_QUEUE
|
|
while (!set.empty()) {
|
|
std::cout << set.top() << std::endl;
|
|
set.pop();
|
|
}
|
|
#else
|
|
for (const auto& item : set) {
|
|
std::cout << item << std::endl;
|
|
}
|
|
#endif
|
|
#endif
|
|
}
|