From cf0afe3223d0e18df33b2fc46c3acf4ab624ba72 Mon Sep 17 00:00:00 2001
From: DovLin <dovis.zhang02@gmail.com>
Date: Fri, 19 Jun 2026 04:30:20 +0800
Subject: [PATCH] [FA]Add lost params in fa varlen func (#28642)

---
 python/sglang/jit_kernel/flash_attention.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/sglang/jit_kernel/flash_attention.py b/python/sglang/jit_kernel/flash_attention.py
index c6a67757d..22c7dc9bc 100644
--- a/python/sglang/jit_kernel/flash_attention.py
+++ b/python/sglang/jit_kernel/flash_attention.py
@@ -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,