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
@@ -42,7 +42,7 @@ def _format_command(cmd):

 def compile_with_raptor(network_path, raptor_onnx_path: Path, output_base: Path,
                        crossbar_size, crossbar_count, core_count=None, pim_merge_scheduler="peft",
-                        cwd=None, verbose=False, reporter=None, timeout_sec=None):
+                        pim_memory_report="none", cwd=None, verbose=False, reporter=None, timeout_sec=None):
    # Define the arguments, with the possibility to set crossbar size and count
    args = [
        network_path,
@@ -56,6 +56,8 @@ def compile_with_raptor(network_path, raptor_onnx_path: Path, output_base: Path,
    ]
    if core_count is not None:
        args.append(f"--core-count={core_count}")
+    if pim_memory_report != "none":
+        args.append(f"--pim-memory-report={pim_memory_report}")
    if verbose:
        args.append("--enable-timing")