From 8b4d7b9519e685b86ec9940de4d1cb4e7f44b13d Mon Sep 17 00:00:00 2001
From: CalamitousFelicitousness <iowasovereign@gmail.com>
Date: Sun, 12 Jul 2026 18:19:30 +0100
Subject: [PATCH] fix(sdnq): end the benchmark cleanly when a faulted cuda
 context is detected

After the correctness section aborts on a sticky kernel fault, the
remaining sections cannot run; flush outputs and exit instead of
cascading into the next cuda call's traceback.
---
 cli/sdnq-attention-benchmark.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/cli/sdnq-attention-benchmark.py b/cli/sdnq-attention-benchmark.py
index 9dd7b21dd..895c7414b 100755
--- a/cli/sdnq-attention-benchmark.py
+++ b/cli/sdnq-attention-benchmark.py
@@ -2694,6 +2694,9 @@ def main():
     print_environment(fp8_result, prep_status, prep_detail, weight_dequant_result=weight_dequant_result)
     if not args.skip_checks and "attention" in sections:
         run_correctness()
+        if report.get("correctness_aborted_after") and not cuda_context_alive():
+            flush_outputs()
+            sys.exit(1)
     if args.skip_bench:
         flush_outputs()
         return