Memory report

src/PIM/Compiler/PimCodeGen.hpp

@@ -1,8 +1,12 @@
 #pragma once
 
-#include "llvm/ADT/DenseMap.h"
+#include "mlir/IR/Operation.h"
 #include "llvm-project/clang/include/clang/Basic/LLVM.h"
+#include "llvm/ADT/DenseMap.h"
 #include "llvm/Support/JSON.h"
+#include "llvm/Support/raw_os_ostream.h"
+
+#include <fstream>
 
 #include "onnx-mlir/Compiler/OMCompilerTypes.h"
 #include "src/Accelerators/PIM/Common/PimCommon.hpp"
@@ -34,6 +38,8 @@ public:
 
   void allocateHost(mlir::ModuleOp moduleOp, mlir::func::FuncOp funcOp);
   void allocateCore(mlir::Operation* op);
+  void report(llvm::raw_ostream& os);
+  void remove(mlir::Value val);
 
   size_t getFirstAvailableAddress() const { return firstAvailableAddress; }
   MemEntry getMemEntry(mlir::Value value) const;
@@ -46,14 +52,28 @@ public:
 
 private:
   llvm::SmallDenseMap<size_t, PimMemory> deviceMem;
+  std::fstream fileReport;
 
 public:
   PimAcceleratorMemory()
-  : hostMem(memEntriesMap) {}
+  : hostMem(memEntriesMap) {
+
+    std::string outputDir = getOutputDir();
+    if (outputDir.empty())
+      return;
+
+    std::string dialectsDir = outputDir + "/reports/";
+    createDirectory(dialectsDir);
+    std::fstream file(dialectsDir + "/memory_report.txt", std::ios::out);
+    fileReport = std::move(file);
+  }
 
   PimMemory& getOrCreateDeviceMem(size_t id);
 
   size_t getValueAddress(mlir::Value value, const StaticValueKnowledge& knowledge = {}) const;
+  void reportHost();
+  void reportCore(size_t coreId);
+  void clean(mlir::Operation* op);
 };
 
 class PimCodeGen {