From 4cbaaae4c70dbb2b3e07844010a381b97217ed76 Mon Sep 17 00:00:00 2001 From: Jie Fang Date: Mon, 6 Jul 2026 14:45:31 +0800 Subject: [PATCH] quick fix for fmha_bwd regressions (#3309) --- .../CuTeDSL/cute/blackwell/kernel/attention/fmha/fmha_bwd.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/examples/python/CuTeDSL/cute/blackwell/kernel/attention/fmha/fmha_bwd.py b/examples/python/CuTeDSL/cute/blackwell/kernel/attention/fmha/fmha_bwd.py index 95d40a3a3..5f02159f2 100644 --- a/examples/python/CuTeDSL/cute/blackwell/kernel/attention/fmha/fmha_bwd.py +++ b/examples/python/CuTeDSL/cute/blackwell/kernel/attention/fmha/fmha_bwd.py @@ -3090,6 +3090,7 @@ def run( window_size_right if window_size_right is None else Int32(window_size_right), workspace, current_stream, + options=f"--opt-level=2" if cutlass.__version__[0:3] == "4.6" else "", ) compilation_time = time.time() - start_time print(f"Compilation time: {compilation_time:.4f} seconds") @@ -3628,3 +3629,4 @@ if __name__ == "__main__": ) print("PASS") +