mirror of
https://github.com/Gericom/teak-llvm.git
synced 2025-06-18 19:15:51 -04:00
aba80d0734
Summary:
This patch adds a benchmarking infrastructure for llvm-libc memory functions.
In a nutshell, the code can benchmark small and large buffers for the memcpy, memset and memcmp functions.
It also produces graphs of size vs latency by running targets of the form `render-libc-{memcpy|memset|memcmp}-benchmark-{small|big}`.
The configurations are provided as JSON files and the benchmark also produces a JSON file.
This file is then parsed and rendered as a PNG file via the `render.py` script (make sure to run `pip3 install matplotlib scipy numpy`).
The script can take several JSON files as input and will superimpose the curves if they are from the same host.
TODO:
- The code benchmarks whatever is available on the host but should be configured to benchmark the -to be added- llvm-libc memory functions.
- Add a README file with instructions and rationale.
- Produce scores to track the performance of the functions over time to allow for regression detection.
Reviewers: sivachandra, ckennelly
Subscribers: mgorny, MaskRay, libc-commits
Tags: #libc-project
Differential Revision: https://reviews.llvm.org/D72516
70 lines
2.2 KiB
C++
70 lines
2.2 KiB
C++
//===-------- Benchmark memcpy implementation -----------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "LibcBenchmark.h"
|
|
#include "LibcMemoryBenchmark.h"
|
|
#include "LibcMemoryBenchmarkMain.h"
|
|
#include "llvm/ADT/StringSwitch.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include <memory>
|
|
|
|
namespace llvm {
|
|
namespace libc_benchmarks {
|
|
|
|
// The context encapsulates the buffers, parameters and the measure.
|
|
struct MemcpyContext : public BenchmarkRunner {
|
|
using FunctionPrototype = void *(*)(void *, const void *, size_t);
|
|
|
|
struct ParameterType {
|
|
uint16_t SrcOffset = 0;
|
|
uint16_t DstOffset = 0;
|
|
};
|
|
|
|
explicit MemcpyContext(const StudyConfiguration &Conf)
|
|
: OD(Conf), SrcBuffer(Conf.BufferSize), DstBuffer(Conf.BufferSize),
|
|
PP(*this) {}
|
|
|
|
// Needed by the ParameterProvider to update the current batch of parameter.
|
|
void Randomize(MutableArrayRef<ParameterType> Parameters) {
|
|
for (auto &P : Parameters) {
|
|
P.DstOffset = OD(Gen);
|
|
P.SrcOffset = OD(Gen);
|
|
}
|
|
}
|
|
|
|
ArrayRef<StringRef> getFunctionNames() const override {
|
|
static std::array<StringRef, 1> kFunctionNames = {"memcpy"};
|
|
return kFunctionNames;
|
|
}
|
|
|
|
BenchmarkResult benchmark(const BenchmarkOptions &Options,
|
|
StringRef FunctionName, size_t Size) override {
|
|
FunctionPrototype Function =
|
|
StringSwitch<FunctionPrototype>(FunctionName).Case("memcpy", &::memcpy);
|
|
return llvm::libc_benchmarks::benchmark(
|
|
Options, PP, [this, Function, Size](ParameterType p) {
|
|
Function(DstBuffer + p.DstOffset, SrcBuffer + p.SrcOffset, Size);
|
|
return DstBuffer + p.DstOffset;
|
|
});
|
|
}
|
|
|
|
private:
|
|
std::default_random_engine Gen;
|
|
OffsetDistribution OD;
|
|
AlignedBuffer SrcBuffer;
|
|
AlignedBuffer DstBuffer;
|
|
SmallParameterProvider<MemcpyContext> PP;
|
|
};
|
|
|
|
std::unique_ptr<BenchmarkRunner> getRunner(const StudyConfiguration &Conf) {
|
|
return std::make_unique<MemcpyContext>(Conf);
|
|
}
|
|
|
|
} // namespace libc_benchmarks
|
|
} // namespace llvm
|