fix gemm segfault
Some checks failed
Validate Operations / config (push) Successful in 1m27s
Validate Operations / build-mlir-cache (push) Successful in 2h0m37s
Validate Operations / validate (push) Failing after 3m41s

print exit signals on validation failure
This commit is contained in:
NiccoloN
2026-03-20 14:00:16 +01:00
parent bb6dcd38a3
commit dbe646ac0d
4 changed files with 52 additions and 5 deletions

View File

@@ -73,10 +73,10 @@ class ProgressReporter:
def suspend(self):
self.suspended = True
self._clear()
sys.stdout.flush()
def resume(self):
self.suspended = False
self._render()
def finish(self):
if self.enabled:
@@ -227,6 +227,7 @@ def validate_network(network_onnx_path, raptor_path, onnx_include_dir,
reporter.log(Fore.CYAN + f"[{model_index}/{model_total}]" + Style.RESET_ALL +
f" {Style.BRIGHT}Validating {network_onnx_path.name}{Style.RESET_ALL}")
failed_with_exception = False
try:
print_stage(reporter, model_index, model_total, network_onnx_path.name, "Compile ONNX")
@@ -287,10 +288,13 @@ def validate_network(network_onnx_path, raptor_path, onnx_include_dir,
reporter.log(Style.BRIGHT + f"Result: {status}" + Style.RESET_ALL)
return passed
except Exception:
failed_with_exception = True
reporter.log(Style.BRIGHT + Fore.RED + "Result: FAIL" + Style.RESET_ALL)
reporter.suspend()
raise
finally:
reporter.log("=" * 72)
if not failed_with_exception:
reporter.log("=" * 72)
if owns_reporter:
reporter.finish()