Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
e3cacb3b
"vllm/vscode:/vscode.git/clone" did not exist on "f58d9b64044e465b85e9280882c738e11b59b2d6"
Commit
e3cacb3b
authored
Aug 01, 2025
by
zhuwenwen
Browse files
update num_rejected_tokens and num_speculative_tokens
parent
7bb28d54
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
8 additions
and
5 deletions
+8
-5
vllm/v1/attention/backends/utils.py
vllm/v1/attention/backends/utils.py
+8
-5
No files found.
vllm/v1/attention/backends/utils.py
View file @
e3cacb3b
...
@@ -45,22 +45,25 @@ class CommonAttentionMetadata:
...
@@ -45,22 +45,25 @@ class CommonAttentionMetadata:
seq_lens_cpu
:
torch
.
Tensor
seq_lens_cpu
:
torch
.
Tensor
"""(batch_size,), the length of each request including both computed tokens
"""(batch_size,), the length of each request including both computed tokens
and newly scheduled tokens"""
and newly scheduled tokens"""
num_computed_tokens_cpu
:
torch
.
Tensor
num_computed_tokens_cpu
:
torch
.
Tensor
"""(batch_size,), the number of computed tokens for each request"""
"""(batch_size,), the number of computed tokens for each request"""
num_reqs
:
int
num_reqs
:
int
"""Number of requests"""
"""Number of requests"""
num_actual_tokens
:
int
num_actual_tokens
:
int
"""Total number of tokens in batch"""
"""Total number of tokens in batch"""
max_query_len
:
int
max_query_len
:
int
"""Longest query in batch"""
"""Longest query in batch"""
num_rejected_tokens
:
list
[
int
]
"""(batch_size,), record the rejected tokens number in cpu and gpu"""
num_speculative_tokens
:
int
"""Number of speculative tokens"""
block_table_tensor
:
torch
.
Tensor
block_table_tensor
:
torch
.
Tensor
slot_mapping
:
torch
.
Tensor
slot_mapping
:
torch
.
Tensor
num_rejected_tokens
:
list
[
int
]
=
None
"""(batch_size,), record the rejected tokens number in cpu and gpu"""
num_speculative_tokens
:
int
=
0
"""Number of speculative tokens"""
M
=
TypeVar
(
"M"
)
M
=
TypeVar
(
"M"
)
...
...
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