Python peft and new summary report

2026-06-05 10:20:02 +02:00
parent 27410207c4
commit 800c0c4316
3 changed files with 9 additions and 3 deletions
@@ -75,6 +75,8 @@ def main():
                    help="Core count to pass to Raptor. Required for PIM validation.")
    ap.add_argument("--pim-merge-scheduler", choices=("peft"), default="peft",
                    help="Scheduler used by the Spatial merge-compute-nodes pass.")
+    ap.add_argument("--pim-memory-report", choices=("none", "summary", "full"), default="none",
+                    help="Emit a human-readable PIM memory planning report during codegen.")
    ap.add_argument("--command-timeout-seconds", type=float, default=1000000.0,
                    help="Per-subprocess timeout in seconds for compiler, runner, and simulator commands.")
    ap.add_argument("--clean", action="store_true",
@@ -142,7 +144,7 @@ def main():
            result = validate_network(
                onnx_path, a.raptor_path, a.onnx_include_dir, simulator_dir,
                crossbar_size=a.crossbar_size, crossbar_count=a.crossbar_count, core_count=a.core_count,
-                pim_merge_scheduler=a.pim_merge_scheduler,
+                pim_merge_scheduler=a.pim_merge_scheduler, pim_memory_report=a.pim_memory_report,
                command_timeout_seconds=a.command_timeout_seconds,
                threshold=a.threshold,
                seed=a.seed,