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

add relu lowering
Some checks failed
Validate Operations / validate-operations (push) Failing after 2h50m56s
Details

Browse Source 
add relu validation
add spatial compute helper
minor refactors
This commit is contained in:
NiccoloN
2026-03-25 11:03:03 +01:00
parent 4e19650b80
commit 742df111e3
29 changed files with 258 additions and 116 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/PIM/PimAccelerator.cpp
View File

@@ -19,6 +19,7 @@
#include "src/Accelerators/PIM/Dialect/Spatial/Transforms/SpatialBufferizableOpInterface.hpp"
#include "src/Accelerators/PIM/Pass/PIMPasses.h"
#include "src/Accelerators/PIM/PimAccelerator.hpp"
#include "src/Compiler/CompilerUtils.hpp"
#define DEBUG_TYPE "PimAccelerator"
@@ -69,13 +70,14 @@ void PimAccelerator::registerDialects(mlir::DialectRegistry& registry) const {
void PimAccelerator::registerPasses(int optLevel) const {
LLVM_DEBUG(llvm::dbgs() << "Registering passes for PIM accelerator\n");
TOTAL_COMPILE_PHASE = 8;
registerPass(createONNXToSpatialPass);
registerPass(createSpatialToGraphvizPass);
registerPass(createSpatialToPimPass);
registerPass(createBufferizePimPass);
registerPass(createConstantFoldingPass);
registerPass(createMaterializeConstantsPass);
registerPass(createVerificationPass);
registerPass(createPimBufferizationPass);
registerPass(createPimConstantFoldingPass);
registerPass(createPimMaterializeConstantsPass);
registerPass(createPimVerificationPass);
registerPass(createEmitPimJsonPass);
}