nnicolosi/Raptor
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Shorter Memory Reporty

Browse Source
This commit is contained in:
ilgeco
2026-05-12 10:38:35 +02:00
parent feaff820e1
commit 26317ea7d0
1 changed files with 18 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/PIM/Compiler/PimCodeGen.cpp
+18 -38
View File
@@ -199,47 +199,27 @@ std::string formatMemory(uint64_t bytes) {
} }
void PimMemory::report(llvm::raw_ostream& file) { void PimMemory::report(llvm::raw_ostream& file) {
// Key: {OpName, Size}, Value: Vector of Addresses uint64_t numAlloca = 0;
// This groups all "memref.alloc" of "1KB" together uint64_t sizeAlloca = 0;
std::vector orderedList(globalMemEntriesMap.begin(), globalMemEntriesMap.end()); uint64_t numGlobal = 0;
std::sort( uint64_t sizeGlobal = 0;
orderedList.begin(), orderedList.end(), [](auto lft, auto rgt) { return lft.second.address < rgt.second.address; }); for (auto& [val, memEntry] : globalMemEntriesMap) {
auto newEnd = std::unique(orderedList.begin(), orderedList.end(), [](auto lft, auto rgt) { if (auto op = val.getDefiningOp()) {
return lft.second.address == rgt.second.address; if (auto allocaOp = dyn_cast<memref::AllocOp>(op)) {
}); numAlloca++;
orderedList.erase(newEnd, orderedList.end()); sizeAlloca += memEntry.size;
std::sort( }
orderedList.begin(), orderedList.end(), [](auto lft, auto rgt) { return lft.second.size < rgt.second.size; });
std::map<std::pair<std::string, uint64_t>, std::vector<uint64_t>> groupedStats;
for (auto& [value, memEntry] : orderedList) { if (auto allocaOp = dyn_cast<memref::GetGlobalOp>(op)) {
std::string opName = "Unknown/BlockArg"; numGlobal++;
if (auto op = value.getDefiningOp()) sizeGlobal += memEntry.size;
opName = op->getName().getStringRef().str(); }
groupedStats[{opName, memEntry.size}].push_back(memEntry.address);
}
file << "--- Memory Usage Report ---\n";
uint64_t totalMemory = 0;
for (auto const& [key, addresses] : groupedStats) {
const std::string& opName = key.first;
uint64_t size = key.second;
file.indent(4) << "Type: " << opName << " [" << formatMemory(size) << "]\n";
file.indent(6) << "Count: " << addresses.size() << "\n";
file.indent(6) << "Total Memory: " << formatMemory(size * addresses.size()) << "\n";
totalMemory += size * addresses.size();
// Optional: Print address range or first/last address to keep it concise
if (!addresses.empty()) {
auto [min, max] = std::minmax_element(addresses.begin(), addresses.end());
file.indent(6) << "Range: " << llvm::format_hex(*min, 10) << " -> " << llvm::format_hex(*max, 10) << "\n";
} }
file << "\n"; file << "\n";
file << "Total Core Memory: " << formatMemory(totalMemory) << "\n"; file << "Total Core Memory: " << formatMemory(totalMemory) << "\n";
} }
file << numAlloca << " " << formatMemory(sizeAlloca) << " " << numGlobal << " " << formatMemory(sizeGlobal) << "\n";
} }
void PimMemory::remove(mlir::Value val) { void PimMemory::remove(mlir::Value val) {
@@ -289,14 +269,14 @@ size_t PimAcceleratorMemory::getValueAddress(mlir::Value value, const StaticValu
void PimAcceleratorMemory::reportHost() { void PimAcceleratorMemory::reportHost() {
llvm::raw_os_ostream os(fileReport); llvm::raw_os_ostream os(fileReport);
os << "Host Memory\n"; os << "Host Memory\t";
hostMem.report(os); hostMem.report(os);
os.flush(); os.flush();
} }
void PimAcceleratorMemory::reportCore(size_t coreId) { void PimAcceleratorMemory::reportCore(size_t coreId) {
llvm::raw_os_ostream os(fileReport); llvm::raw_os_ostream os(fileReport);
os << "Core " << coreId << " Memory\n"; os << "Core " << coreId << " Memory\t";
deviceMem.at(coreId).report(os); deviceMem.at(coreId).report(os);
os.flush(); os.flush();
} }