mirror of
https://github.com/vladmandic/automatic
synced 2026-09-14 10:38:42 +02:00
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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user