Skip to content

Commit 27b8017

Browse files
authored
[FIX][Intel GPU]fix ipex flash_attn_varlen_func api missing parameter (#20348)
Signed-off-by: Kunshang Ji <kunshang.ji@intel.com>
1 parent 9ec1e30 commit 27b8017

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

vllm/_ipex_ops.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -270,6 +270,7 @@ def flash_attn_varlen_func(
270270
q_descale=None,
271271
k_descale=None,
272272
v_descale=None,
273+
num_splits=0,
273274
):
274275
if cu_seqlens_k is None:
275276
# cu_seqlens_k is not used in ipex kernel.

0 commit comments

Comments
 (0)