[XPU]Fix flash_attn_varlen_func interface on xpu (#22350)

Signed-off-by: Kunshang Ji <kunshang.ji@intel.com>
This commit is contained in:
Kunshang Ji
2025-08-07 10:28:11 +08:00
committed by GitHub
parent 1dc8a70b6d
commit 6b47ef24de

View File

@ -271,6 +271,7 @@ class ipex_ops:
k_descale=None,
v_descale=None,
num_splits=0,
s_aux: Optional[torch.Tensor] = None,
):
if cu_seqlens_k is None:
# cu_seqlens_k is not used in ipex kernel.