From 423904d7172ce00329c9f24f38b46f9569999409 Mon Sep 17 00:00:00 2001 From: Linfeng Zheng Date: Wed, 3 Jun 2026 08:55:57 +0800 Subject: [PATCH] [CLI] Recover fmha perf (#3291) * [CLI] Recover fmha perf * [CLI] enable options for a certain version --- .../python/CuTeDSL/cute/blackwell/kernel/attention/fmha/fmha.py | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/python/CuTeDSL/cute/blackwell/kernel/attention/fmha/fmha.py b/examples/python/CuTeDSL/cute/blackwell/kernel/attention/fmha/fmha.py index f9ec6acd0..39ac474d5 100644 --- 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(