From 90b2f638d9929a1bd676a84e97dca4b7ec3806b5 Mon Sep 17 00:00:00 2001
From: geruome <2665633561@qq.com>
Date: Wed, 13 May 2026 21:33:53 +0800
Subject: [PATCH] fix

---
 flash_attn/cute/flash_fwd_sm90.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/flash_attn/cute/flash_fwd_sm90.py b/flash_attn/cute/flash_fwd_sm90.py
index bbbe7a063d9..23f92181166 100644
--- a/flash_attn/cute/flash_fwd_sm90.py
+++ b/flash_attn/cute/flash_fwd_sm90.py
@@ -888,6 +888,7 @@ def load(
                             batch_idx,
                             head_idx,
                             m_block,
+                            seqlen,
                             kv_producer_state,
                             tma_load_K_fn,
                             tma_load_V_fn,