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,
     )