From e3d6d6eebf992e0f9fcbf4547de753c839a3600e Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Sat, 23 May 2026 02:25:46 +0000
Subject: [PATCH] diag: add CUDA_LAUNCH_BLOCKING for crash debug

---
 tests/unit/test_fmha_v3_stage_c.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/unit/test_fmha_v3_stage_c.py b/tests/unit/test_fmha_v3_stage_c.py
index 9f638dda..681b9cdd 100644
--- a/tests/unit/test_fmha_v3_stage_c.py
+++ b/tests/unit/test_fmha_v3_stage_c.py
@@ -423,6 +423,8 @@ class FmhaV3StageCMulti:
 
 
 def test():
+    import os
+    os.environ['CUDA_LAUNCH_BLOCKING'] = '1'
     torch.manual_seed(42)
     for n in [128]:
         torch.manual_seed(42)