[FA]Add lost params in fa varlen func (#28642)

2026-06-18 13:30:20 -07:00
parent 66a7fd5c0b
commit cf0afe3223
1 changed files with 2 additions and 0 deletions
@@ -230,6 +230,7 @@ def flash_attn_varlen_func(
    softcap=0.0,
    num_splits=1,
    pack_gqa=None,
+    only_qv=False,
    sm_margin=0,
    return_softmax_lse=False,
    sinks=None,
@@ -262,6 +263,7 @@ def flash_attn_varlen_func(
            softcap=softcap,
            num_splits=num_splits,
            pack_gqa=pack_gqa,
+            only_qv=only_qv,
            sm_margin=sm_margin,
            return_softmax_lse=return_softmax_lse,
            sinks=sinks,