mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-28 18:16:06 +00:00
30b93db547
Adds compile time flag -mllvm -memprof-histogram and runtime flag histogram=true|false to turn Histogram collection on and off. The -memprof-histogram flag relies on -memprof-use-callbacks=true to work. Updates shadow mapping logic in histogram mode from having one 8 byte counter for 64 bytes, to 1 byte for 8 bytes, capped at 255. Only supports this granularity as of now. Updates the RawMemprofReader and serializing MemoryInfoBlocks to binary format, including changing to a new version of the raw binary format from version 3 to version 4. Updates creating MemoryInfoBlocks with and without Histograms. When two MemoryInfoBlocks are merged, AccessCounts are summed up and the shorter Histogram is removed. Adds a memprof_histogram test case. Initial commit for adding AccessCountHistograms up until RawProfile for memprof
49 lines
1.7 KiB
C++
49 lines
1.7 KiB
C++
//===-- memprof_mibmap.cpp -----------------------------------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file is a part of MemProfiler, a memory profiler.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "memprof_mibmap.h"
|
|
#include "profile/MemProfData.inc"
|
|
#include "sanitizer_common/sanitizer_allocator_internal.h"
|
|
#include "sanitizer_common/sanitizer_mutex.h"
|
|
|
|
namespace __memprof {
|
|
using ::llvm::memprof::MemInfoBlock;
|
|
|
|
void InsertOrMerge(const uptr Id, const MemInfoBlock &Block, MIBMapTy &Map) {
|
|
MIBMapTy::Handle h(&Map, static_cast<uptr>(Id), /*remove=*/false,
|
|
/*create=*/true);
|
|
if (h.created()) {
|
|
LockedMemInfoBlock *lmib =
|
|
(LockedMemInfoBlock *)InternalAlloc(sizeof(LockedMemInfoBlock));
|
|
lmib->mutex.Init();
|
|
lmib->mib = Block;
|
|
*h = lmib;
|
|
} else {
|
|
LockedMemInfoBlock *lmib = *h;
|
|
SpinMutexLock lock(&lmib->mutex);
|
|
uintptr_t ShorterHistogram;
|
|
if (Block.AccessHistogramSize > lmib->mib.AccessHistogramSize)
|
|
ShorterHistogram = lmib->mib.AccessHistogram;
|
|
else
|
|
ShorterHistogram = Block.AccessHistogram;
|
|
|
|
lmib->mib.Merge(Block);
|
|
// The larger histogram is kept and the shorter histogram is discarded after
|
|
// adding the counters to the larger historam. Free only the shorter
|
|
// Histogram
|
|
if (Block.AccessHistogramSize > 0 || lmib->mib.AccessHistogramSize > 0)
|
|
InternalFree((void *)ShorterHistogram);
|
|
}
|
|
}
|
|
|
|
} // namespace __memprof
|