From 6b47ef24de3d3b4f551aca0bc21b9f16f3d21b6a Mon Sep 17 00:00:00 2001
From: Kunshang Ji <kunshang.ji@intel.com>
Date: Thu, 7 Aug 2025 10:28:11 +0800
Subject: [PATCH] [XPU]Fix `flash_attn_varlen_func` interface on xpu (#22350)

Signed-off-by: Kunshang Ji <kunshang.ji@intel.com>
---
 vllm/_ipex_ops.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/vllm/_ipex_ops.py b/vllm/_ipex_ops.py
index 7533bf5ef..79e3e448c 100644
--- a/vllm/_ipex_ops.py
+++ b/vllm/_ipex_ops.py
@@ -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.