Print aie-assign-buffer-addresses errors to aiecc stderr (#1427)

Xilinx · Apr 29, 2024 · 774f25b · 774f25b
1 parent 2f133fe
commit 774f25b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/python/compiler/aiecc/main.py b/python/compiler/aiecc/main.py
@@ -996,7 +996,7 @@ async def run_flow(self):
 
             file_with_addresses = self.prepend_tmp("input_with_addresses.mlir")
             if opts.basic_alloc_scheme:
-                do_run(
+                r = do_run(
                     [
                         "aie-opt",
                         "--aie-assign-buffer-addresses=basic-alloc",
@@ -1006,7 +1006,7 @@ async def run_flow(self):
                     ],
                 )
             else:
-                do_run(
+                r = do_run(
                     [
                         "aie-opt",
                         "--aie-assign-buffer-addresses",
@@ -1015,6 +1015,10 @@ async def run_flow(self):
                         file_with_addresses,
                     ],
                 )
+            if r.returncode != 0:
+                print("Error encountered while assigning buffer addresses. Exiting...")
+                print(r.stderr, file=sys.stderr)
+                sys.exit(r.returncode)
 
             cores = generate_cores_list(await read_file_async(file_with_addresses))
             t = do_run(