From bab9379c376c0922bbd5da0ab6c12d4c2872784e Mon Sep 17 00:00:00 2001
From: Ivan Butygin <ivan.butygin@gmail.com>
Date: Wed, 25 Jun 2025 10:33:10 +0200
Subject: [PATCH] fix extend

Signed-off-by: Ivan Butygin <ivan.butygin@gmail.com>
---
 python/sglang/srt/layers/attention/wave_ops/extend_attention.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/sglang/srt/layers/attention/wave_ops/extend_attention.py b/python/sglang/srt/layers/attention/wave_ops/extend_attention.py
index 0c7c70eb7502..55c4bb66dc9c 100644
--- a/python/sglang/srt/layers/attention/wave_ops/extend_attention.py
+++ b/python/sglang/srt/layers/attention/wave_ops/extend_attention.py
@@ -160,6 +160,7 @@ def extend_attention_wave(
         qo_indptr,
         kv_indptr,
         kv_indices,
+        max_seq_len,
         output,
     )