even faster on pimcomp models

2026-07-31 21:15:28 +02:00
parent 9ca1a0ed9f
commit f4a3b012cc
49 changed files with 1923 additions and 583 deletions
@@ -281,6 +281,7 @@ void addPassesPim(OwningOpRef<ModuleOp>& module,
  verifyExplicitPimCoreCount();

  if (pimOnlyCodegen) {
+    pm.addPass(createPimInstructionSelectionPass());
    pm.addPass(createPimLocalMemoryPlanningPass());
    pm.addPass(createPimVerificationPass());
    pm.addPass(createEmitPimCodePass());
@@ -315,6 +316,8 @@ void addPassesPim(OwningOpRef<ModuleOp>& module,
    pm.addPass(mlir::createLowerAffinePass());
    pm.addPass(createPimHostConstantFoldingPass());
    pm.addPass(createMessagePass("Pim host constants folded"));
+    pm.addPass(createPimInstructionSelectionPass());
+    pm.addPass(createMessagePass("Pim instructions selected"));
    pm.addPass(createPimLocalMemoryPlanningPass());
    pm.addPass(createMessagePass("Pim local memory planned"));
    pm.addPass(createPimVerificationPass());