From 8bb5385961c9a6abd23ecc97c6a612385ba1dbc8 Mon Sep 17 00:00:00 2001 From: NiccoloN Date: Thu, 26 Feb 2026 17:53:25 +0100 Subject: [PATCH] add host core pim codegen --- src/PIM/Compiler/PimCodeGen.cpp | 9 +++++++++ src/PIM/Conversion/SpatialToPIM/SpatialToPIMPass.cpp | 2 +- 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/src/PIM/Compiler/PimCodeGen.cpp b/src/PIM/Compiler/PimCodeGen.cpp index 25d0a6f..e60d3a7 100644 --- a/src/PIM/Compiler/PimCodeGen.cpp +++ b/src/PIM/Compiler/PimCodeGen.cpp @@ -494,6 +494,15 @@ OnnxMlirCompilerErrorCodes compileModuleToPIMJSON(ModuleOp& moduleOp, std::strin memoryFileStream.write(memoryBuffer.data(), memoryBuffer.size()); memoryFileStream.close(); + auto outputHostCorePath = outputDirPath + "/core_0.json"; + llvm::raw_fd_ostream hostFileStream(outputHostCorePath, errorCode); + if (errorCode) { + llvm::errs() << "Error while opening host core file `" << outputHostCorePath << "`: " << errorCode.message() << '\n'; + return InvalidOutputFileAccess; + } + hostFileStream << "[]"; + hostFileStream.close(); + size_t coreCount = 0; for (auto coreOp : funcOp.getOps()) { auto coreId = coreOp.getCoreId(); diff --git a/src/PIM/Conversion/SpatialToPIM/SpatialToPIMPass.cpp b/src/PIM/Conversion/SpatialToPIM/SpatialToPIMPass.cpp index 11d9a8a..c4ed4d0 100644 --- a/src/PIM/Conversion/SpatialToPIM/SpatialToPIMPass.cpp +++ b/src/PIM/Conversion/SpatialToPIM/SpatialToPIMPass.cpp @@ -25,7 +25,7 @@ namespace onnx_mlir { namespace pim { void SpatialToPIMPass::runOnOperation() { - coreId = 0; + coreId = 1; ModuleOp moduleOp = getOperation(); MLIRContext* ctx = moduleOp.getContext();