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.
This commit is contained in:
CalamitousFelicitousness
2026-07-12 18:19:30 +01:00
parent 49ee0dc647
commit 8b4d7b9519
+3
View File
@@ -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