nnicolosi/Raptor
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add host core pim codegen

Browse Source
This commit is contained in:
NiccoloN
2026-02-26 17:53:25 +01:00
parent f007b6e574
commit 8bb5385961
2 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/PIM/Compiler/PimCodeGen.cpp
View File

@@ -494,6 +494,15 @@ OnnxMlirCompilerErrorCodes compileModuleToPIMJSON(ModuleOp& moduleOp, std::strin
memoryFileStream.write(memoryBuffer.data(), memoryBuffer.size()); memoryFileStream.write(memoryBuffer.data(), memoryBuffer.size());
memoryFileStream.close(); 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; size_t coreCount = 0;
for (auto coreOp : funcOp.getOps<pim::PimCoreOp>()) { for (auto coreOp : funcOp.getOps<pim::PimCoreOp>()) {
auto coreId = coreOp.getCoreId(); auto coreId = coreOp.getCoreId();

2
src/PIM/Conversion/SpatialToPIM/SpatialToPIMPass.cpp
View File

@@ -25,7 +25,7 @@ namespace onnx_mlir {
namespace pim { namespace pim {
void SpatialToPIMPass::runOnOperation() { void SpatialToPIMPass::runOnOperation() {
coreId = 0; coreId = 1;
ModuleOp moduleOp = getOperation(); ModuleOp moduleOp = getOperation();
MLIRContext* ctx = moduleOp.getContext(); MLIRContext* ctx = moduleOp.getContext();