diag: add CUDA_LAUNCH_BLOCKING for crash debug

2026-05-23 02:25:46 +00:00
parent 1ee99cbef6
commit e3d6d6eebf
1 changed files with 2 additions and 0 deletions
--- 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)