Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
change
sglang
Commits
27e327b4
Unverified
Commit
27e327b4
authored
Jun 03, 2025
by
pansicheng
Committed by
GitHub
Jun 02, 2025
Browse files
fix new_page_count_next_decode (#6671)
parent
ff00895c
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
1 deletion
+3
-1
python/sglang/srt/managers/schedule_batch.py
python/sglang/srt/managers/schedule_batch.py
+3
-1
No files found.
python/sglang/srt/managers/schedule_batch.py
View file @
27e327b4
...
...
@@ -1333,7 +1333,9 @@ class ScheduleBatch(ScheduleBatchDisaggregationDecodeMixin):
page_size
=
self
.
token_to_kv_pool_allocator
.
page_size
if
page_size
==
1
:
return
len
(
self
.
reqs
)
return
sum
(
1
for
req
in
self
.
reqs
if
req
.
seqlen
%
page_size
==
0
)
# In the decoding phase, the length of a request's KV cache should be
# the total length of the request minus 1
return
sum
(
1
for
req
in
self
.
reqs
if
(
req
.
seqlen
-
1
)
%
page_size
==
0
)
def
check_decode_mem
(
self
,
buf_multiplier
=
1
):
tokens_required
=
(
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment