Raptor sync wait

2026-08-06 14:32:46 +02:00
parent a963009855
commit a39fdba366
48 changed files with 3357 additions and 96 deletions
@@ -330,6 +330,7 @@ void addPassesPim(OwningOpRef<ModuleOp>& module,
                  EmissionTargetType& emissionTarget,
                  std::string outputNameNoExt) {
  verifyExplicitPimCoreCount();
+  verifyPimPipelineStages();
  spatial::SchedulingTarget schedulingTarget = getPimSchedulingTarget();
  spatial::SpatialTargetResources targetResources = getPimSpatialTargetResources(schedulingTarget);

@@ -354,7 +355,7 @@ void addPassesPim(OwningOpRef<ModuleOp>& module,
    pm.addPass(createTrivialGraphComputeMergePass(
      schedulingTarget.residentWeightCapacity, exportStage));
    pm.addPass(spatial::createScheduleAndRealizeSpatialPass(
-      schedulingTarget, exportStage));
+      schedulingTarget, exportStage, pipelineStages.getValue()));
    pm.addPass(createMessagePass("Onnx lowered to Spatial"));
  }