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
eebf00cb
Unverified
Commit
eebf00cb
authored
Oct 30, 2025
by
Li, Jiang
Committed by
GitHub
Oct 30, 2025
Browse files
[Bugfix][CPU] Fix MRoPE dispatch on the CPU backend (#27800)
Signed-off-by:
jiang1.li
<
jiang1.li@intel.com
>
parent
9956aae4
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
9 additions
and
0 deletions
+9
-0
vllm/model_executor/layers/rotary_embedding/mrope.py
vllm/model_executor/layers/rotary_embedding/mrope.py
+9
-0
No files found.
vllm/model_executor/layers/rotary_embedding/mrope.py
View file @
eebf00cb
...
@@ -357,6 +357,15 @@ class MRotaryEmbedding(RotaryEmbeddingBase):
...
@@ -357,6 +357,15 @@ class MRotaryEmbedding(RotaryEmbeddingBase):
key
=
torch
.
cat
((
key_rot
,
key_pass
),
dim
=-
1
).
reshape
(
key_shape
)
key
=
torch
.
cat
((
key_rot
,
key_pass
),
dim
=-
1
).
reshape
(
key_shape
)
return
query
,
key
return
query
,
key
def
forward_cpu
(
self
,
positions
:
torch
.
Tensor
,
query
:
torch
.
Tensor
,
key
:
torch
.
Tensor
|
None
=
None
,
offsets
:
torch
.
Tensor
|
None
=
None
,
)
->
tuple
[
torch
.
Tensor
,
torch
.
Tensor
|
None
]:
return
self
.
forward_native
(
positions
,
query
,
key
,
offsets
)
@
staticmethod
@
staticmethod
def
get_next_input_positions
(
def
get_next_input_positions
(
mrope_position_delta
:
int
,
mrope_position_delta
:
int
,
...
...
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