From 511accb259be2650fbcd0c5199a0059bcabf59b1 Mon Sep 17 00:00:00 2001
From: Erik Lundell <erik.lundell@arm.com>
Date: Thu, 11 Apr 2024 16:37:18 +0200
Subject: [PATCH] Capture output of Vela and print on error

Change-Id: I0443a6ab26766a51511d9e4ea532fc8e76836ede
---
 backends/arm/arm_vela.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/backends/arm/arm_vela.py b/backends/arm/arm_vela.py
index 0d0f0eb0377..f387672b7b4 100644
--- a/backends/arm/arm_vela.py
+++ b/backends/arm/arm_vela.py
@@ -43,7 +43,14 @@ def vela_compile(tosa_graph, args: List[str]):
 
         # invoke vela
         vela_command = f"cd {tmpdir}; vela {' '.join(args)} {tosaname}"
-        subprocess.run([vela_command], shell=True, check=True)
+        try:
+            subprocess.run([vela_command], shell=True, check=True, capture_output=True)
+        except subprocess.CalledProcessError as process_error:
+            raise RuntimeError(
+                f"Vela compiler ('{vela_command}') failed with error:\n \
+                                     {process_error.stderr.decode()}\n \
+                                      Stdout:\n{process_error.stdout.decode()}"
+            )
 
         np_path = os.path.join(tmpdir, "output", "out_sg0_vela.npz")
         blocks = b""