diff options
| author | udovichenko-r <udovichenko-r@yandex-team.ru> | 2022-07-04 14:16:38 +0300 |
|---|---|---|
| committer | udovichenko-r <udovichenko-r@yandex-team.ru> | 2022-07-04 14:16:38 +0300 |
| commit | 5fe1c2b2d90b4ddbd7d1683191a48851363cf53d (patch) | |
| tree | c413b43fb69611ff1185ead7813a8e0973dca3dc /library/cpp/balloc/lib/alloc_stats.cpp | |
| parent | f9651ab5ad67347bf06d6d0789b5d6eb31a7b2cc (diff) | |
| download | ydb-5fe1c2b2d90b4ddbd7d1683191a48851363cf53d.tar.gz | |
[KIKIMR-15108] Add perf programs to build
ref:8f081efde09627da76e52231d32a83e34b78c1e4
Diffstat (limited to 'library/cpp/balloc/lib/alloc_stats.cpp')
| -rw-r--r-- | library/cpp/balloc/lib/alloc_stats.cpp | 106 |
1 files changed, 106 insertions, 0 deletions
diff --git a/library/cpp/balloc/lib/alloc_stats.cpp b/library/cpp/balloc/lib/alloc_stats.cpp new file mode 100644 index 0000000000..3481cad56c --- /dev/null +++ b/library/cpp/balloc/lib/alloc_stats.cpp @@ -0,0 +1,106 @@ +#include <library/cpp/balloc/lib/alloc_stats.h> + +#include <util/system/compiler.h> +#include <atomic> + + +namespace NAllocStats { + +struct TThreadAllocStats { + i64 CurrSize = 0; + i64 MaxSize = 0; +}; + +struct TGlobalAllocStats { + std::atomic<ui64> LiveLock = {0}; + std::atomic<ui64> Mmap = {0}; +}; + +#if defined(_unix_) && !defined(_darwin_) + +__thread bool isEnabled = false; + +bool IsEnabled() noexcept { + return isEnabled; +} + +void EnableAllocStats(bool enable) noexcept { + isEnabled = enable; +} + +__thread TThreadAllocStats threadAllocStats; + +void IncThreadAllocStats(i64 size) noexcept { + threadAllocStats.CurrSize += size; + if (Y_UNLIKELY(threadAllocStats.CurrSize > threadAllocStats.MaxSize)) { + threadAllocStats.MaxSize = threadAllocStats.CurrSize; + } +} + +void DecThreadAllocStats(i64 size) noexcept { + threadAllocStats.CurrSize -= size; +} + +void ResetThreadAllocStats() noexcept { + threadAllocStats.CurrSize = 0; + threadAllocStats.MaxSize = 0; +} + +i64 GetThreadAllocMax() noexcept { + return threadAllocStats.MaxSize; +} + +#else // _unix_ && ! _darwin_ + +bool IsEnabled() noexcept { + return false; +} +void EnableAllocStats(bool /*enable*/) noexcept { +} +void IncThreadAllocStats(i64 /*size*/) noexcept { +} +void DecThreadAllocStats(i64 /*size*/) noexcept { +} +void ResetThreadAllocStats() noexcept { +} +i64 GetThreadAllocMax() noexcept { + return 0; +} + +#endif // _unix_ && ! _darwin_ + + +#if defined(_x86_64_) || defined(_i386_) + static constexpr size_t CACHE_LINE_SIZE = 64; +#elif defined(_arm64_) || defined(_ppc64_) + static constexpr size_t CACHE_LINE_SIZE = 128; +#else + static constexpr size_t CACHE_LINE_SIZE = 256; // default large enough +#endif + +template <typename T> +struct alignas(sizeof(T)) TCacheLineDoublePaddedAtomic { + char Prefix[CACHE_LINE_SIZE - sizeof(T)]; + T Value; + char Postfix[CACHE_LINE_SIZE - sizeof(T)]; +}; + +TCacheLineDoublePaddedAtomic<TGlobalAllocStats> GlobalCounters; + +void IncLiveLockCounter() noexcept { + GlobalCounters.Value.LiveLock.fetch_add(1, std::memory_order_seq_cst); +} + +ui64 GetLiveLockCounter() noexcept { + return GlobalCounters.Value.LiveLock.load(std::memory_order_acquire); +} + +void IncMmapCounter(ui64 amount) noexcept { + GlobalCounters.Value.Mmap.fetch_add(amount, std::memory_order_seq_cst); +} + +ui64 GetMmapCounter() noexcept { + return GlobalCounters.Value.Mmap.load(std::memory_order_acquire); +} + +} // namespace NAllocStats |