Arm backend: aot_arm_compiler.py: Create output dir (#11282)

martinlsm · Martin Lindström · web-flow · commit a1ce09aeec96 · 2025-06-02T14:28:58.000+02:00
Create output directory for the generated BundledProgram if it does not
already exist. This will save the user time by not having to rerun the
program due to it complaining that the directory does not exist.

Co-authored-by: Martin Lindström &lt;Martin.Lindstroem@arm.com&gt;
diff --git a/examples/arm/aot_arm_compiler.py b/examples/arm/aot_arm_compiler.py
@@ -668,6 +668,8 @@ def save_bpte_program(exec_prog, original_model: torch.nn.Module, output_name: s
         )
 
     # Generate BundledProgram
+    output_dir = os.path.dirname(output_name)
+    os.makedirs(output_dir, exist_ok=True)
     save_bundled_program(exec_prog, method_test_suites, output_name)
 
 

Original file line number	Diff line number	Diff line change
`@@ -668,6 +668,8 @@ def save_bpte_program(exec_prog, original_model: torch.nn.Module, output_name: s`
`668`	`668`	`)`
`669`	`669`
`670`	`670`	`# Generate BundledProgram`
	`671`	`+ output_dir = os.path.dirname(output_name)`
	`672`	`+ os.makedirs(output_dir, exist_ok=True)`
`671`	`673`	`save_bundled_program(exec_prog, method_test_suites, output_name)`
`672`	`674`
`673`	`675`