CodexWorkaround

2026-06-08 11:33:36 +02:00
parent aec80529ca
commit 75fb70712f
184 changed files with 1127 additions and 7 deletions
@@ -41,7 +41,8 @@ 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",
-                        pim_memory_report="none", cwd=None, verbose=False, reporter=None, timeout_sec=None):
+                        pim_memory_report="none", raptor_extra_args=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,
@@ -57,6 +58,8 @@ def compile_with_raptor(network_path, raptor_onnx_path: Path, output_base: Path,
        args.append(f"--core-count={core_count}")
    if pim_memory_report != "none":
        args.append(f"--pim-memory-report={pim_memory_report}")
+    if raptor_extra_args:
+        args.extend(str(arg) for arg in raptor_extra_args)
    if verbose:
        args.append("--enable-timing")