add PIM accelerator

2026-02-24 15:09:18 +01:00
parent b24a0df8d7
commit a6e928bdd7
67 changed files with 9109 additions and 1 deletions
@@ -0,0 +1,19 @@
+#pragma once
+
+#include "mlir/IR/BuiltinOps.h"
+#include "mlir/Pass/PassManager.h"
+#include "mlir/Pass/PassRegistry.h"
+
+#include "onnx-mlir/Compiler/OMCompilerTypes.h"
+
+namespace onnx_mlir {
+
+void addPassesPim(mlir::OwningOpRef<mlir::ModuleOp>& module,
+                  mlir::PassManager& pm,
+                  EmissionTargetType& emissionTarget,
+                  std::string outputNameNoExt);
+
+int compileModuleToPIMJSON(const mlir::OwningOpRef<mlir::ModuleOp>& moduleOpRef,
+                           std::string& outputDirName);
+
+} // namespace onnx_mlir