better memory report

capped vector allocations at u32::MAX in rust simulator
2026-06-03 13:48:42 +02:00
parent 3c2667f11e
commit 501e6c76f3
6 changed files with 88 additions and 38 deletions
@@ -6,6 +6,7 @@
 #include "llvm/ADT/SmallVector.h"
 #include "llvm/ADT/StringRef.h"

+#include <cstdint>
 #include <string>

 #include "src/Accelerators/PIM/Common/IR/WeightUtils.hpp"
@@ -17,7 +18,12 @@ struct WeightFileRequest {
  llvm::SmallVector<ResolvedWeightView, 8> weights;
 };

-llvm::DenseMap<size_t, llvm::SmallVector<std::string, 8>>
+struct WeightEmissionResult {
+  llvm::DenseMap<size_t, llvm::SmallVector<std::string, 8>> mapCoreWeightToFileName;
+  uint64_t totalWeightBytes = 0;
+};
+
+WeightEmissionResult
 createAndPopulateWeightFolder(llvm::ArrayRef<WeightFileRequest> requests, llvm::StringRef outputDirPath);

 } // namespace onnx_mlir