[CLI] Recover fmha perf (#3291)

* [CLI] Recover fmha perf * [CLI] enable options for a certain version
2026-06-29 02:47:05 +00:00 · 2026-06-03 08:55:57 +08:00
parent 25e252bdce
commit 423904d717
1 changed files with 1 additions and 0 deletions
--- a/examples/python/CuTeDSL/cute/blackwell/kernel/attention/fmha/fmha.py
+++ b/examples/python/CuTeDSL/cute/blackwell/kernel/attention/fmha/fmha.py
@@ -4003,6 +4003,7 @@ def run(
        total_softmax_count,
        current_stream,
        use_pdl,
+        options=f"--opt-level=2" if cutlass.__version__[0:3]=="4.6" else "",
    )
    if stats_skip_softmax_ratio:
        print(