binary pim code for reduced memory usage

fast pim code emission

src/PIM/Pass/CMakeLists.txt

@@ -7,7 +7,7 @@ add_pim_library(OMPimPasses
   PimCodegen/HostConstantFolding/Patterns/Subview.cpp
   PimCodegen/MaterializeHostConstantsPass.cpp
   PimCodegen/VerificationPass.cpp
-  PimCodegen/EmitPimJsonPass.cpp
+  PimCodegen/EmitPimCodePass.cpp
 
   EXCLUDE_FROM_OM_LIBS
 

src/PIM/Pass/PIMPasses.h

@@ -25,7 +25,7 @@ std::unique_ptr<mlir::Pass> createPimMaterializeHostConstantsPass();
 
 std::unique_ptr<mlir::Pass> createPimVerificationPass();
 
-std::unique_ptr<mlir::Pass> createEmitPimJsonPass();
+std::unique_ptr<mlir::Pass> createEmitPimCodePass();
 
 std::unique_ptr<mlir::Pass> createMessagePass(std::string message);
 

src/PIM/Pass/PimCodegen/EmitPimCodePass.cpp

@@ -0,0 +1,36 @@
+#include "mlir/Pass/Pass.h"
+
+#include "Common/PimCommon.hpp"
+#include "Compiler/PimCodeGen.hpp"
+
+using namespace mlir;
+
+namespace onnx_mlir {
+
+namespace {
+
+struct EmitPimCodePass : PassWrapper<EmitPimCodePass, OperationPass<ModuleOp>> {
+  MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(EmitPimCodePass);
+  StringRef getArgument() const override { return "emit-pim-code-pass"; }
+  StringRef getDescription() const override { return "Emit PIM simulator code artifacts"; }
+
+  EmitPimCodePass() {}
+  EmitPimCodePass(const EmitPimCodePass& pass) {}
+
+  void runOnOperation() override {
+    ModuleOp moduleOp = getOperation();
+
+    std::string pimDir = getOutputDir() + "/pim";
+    createDirectory(pimDir);
+
+    int compiler_error_code = compileToPimCode(moduleOp, pimDir);
+    if (compiler_error_code != CompilerSuccess)
+      signalPassFailure();
+  }
+};
+
+} // namespace
+
+std::unique_ptr<Pass> createEmitPimCodePass() { return std::make_unique<EmitPimCodePass>(); }
+
+} // namespace onnx_mlir

src/PIM/Pass/PimCodegen/EmitPimJsonPass.cpp

@@ -1,36 +0,0 @@
-#include "mlir/Pass/Pass.h"
-
-#include "Common/PimCommon.hpp"
-#include "Compiler/PimCodeGen.hpp"
-
-using namespace mlir;
-
-namespace onnx_mlir {
-
-namespace {
-
-struct EmitPimJsonPass : PassWrapper<EmitPimJsonPass, OperationPass<ModuleOp>> {
-  MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(EmitPimJsonPass);
-  StringRef getArgument() const override { return "emit-pim-json-pass"; }
-  StringRef getDescription() const override { return "Emit json code for the pim simulators"; }
-
-  EmitPimJsonPass() {}
-  EmitPimJsonPass(const EmitPimJsonPass& pass) {}
-
-  void runOnOperation() override {
-    ModuleOp moduleOp = getOperation();
-
-    std::string pimDir = getOutputDir() + "/pim";
-    createDirectory(pimDir);
-
-    int compiler_error_code = compileToPimJson(moduleOp, pimDir);
-    if (compiler_error_code != CompilerSuccess)
-      signalPassFailure();
-  }
-};
-
-} // namespace
-
-std::unique_ptr<Pass> createEmitPimJsonPass() { return std::make_unique<EmitPimJsonPass>(); }
-
-} // namespace onnx_mlir