add PIM accelerator

src/PIM/Compiler/PimCompilerUtils.cpp

@@ -0,0 +1,56 @@
+#include "mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h"
+#include "mlir/Dialect/Bufferization/Transforms/BufferViewFlowAnalysis.h"
+#include "mlir/Dialect/MemRef/IR/MemRef.h"
+#include "mlir/IR/BuiltinTypes.h"
+#include "mlir/Transforms/Passes.h"
+#include "src/Accelerators/PIM/Compiler/PimCompilerOptions.hpp"
+#include "src/Accelerators/PIM/Compiler/PimCompilerUtils.hpp"
+#include "src/Accelerators/PIM/Dialect/PIM/PimOps.hpp"
+#include "src/Accelerators/PIM/Pass/PimPasses.hpp"
+#include "src/Compiler/CompilerPasses.hpp"
+
+#include "llvm/Support/JSON.h"
+
+#include <cassert>
+#include <cstddef>
+
+#define DEBUG_TYPE "PimCompilerUtils"
+
+using namespace mlir;
+using namespace onnx_mlir;
+
+namespace onnx_mlir {
+
+void addPassesPim(OwningOpRef<ModuleOp>& module,
+                  PassManager& pm,
+                  EmissionTargetType& emissionTarget,
+                  std::string outputNameNoExt) {
+
+  if (pimOnlyCodegen) {
+    // Skip all the lowering passes and directly generate code for PIM.
+    return;
+  }
+
+  if (emissionTarget >= EmitONNXIR)
+    addONNXToMLIRPasses(pm, /*target CPU*/ false);
+
+  if (pimEmissionTarget >= EmitSpatial) {
+    pm.addPass(createONNXToSpatialPass());
+    // pm.addPass(createCountInstructionPass());
+    pm.addPass(createMessagePass("ONNX lowered to SPATIAL"));
+  }
+
+  if (pimEmissionTarget >= EmitPim) {
+    pm.addPass(createSpatialToPIMPass());
+    // pm.addPass(createCountInstructionPass());
+    pm.addPass(createMessagePass("SPATIAL lowered to PIM"));
+  }
+
+  if (pimEmissionTarget >= EmitPimBufferized) {
+    pm.addPass(createBufferizePimPass());
+    // pm.addPass(createCountInstructionPass());
+    pm.addPass(createMessagePass("PIM bufferized"));
+  }
+}
+
+} // namespace onnx_mlir