[XPU] follow cuda path for mrope on XPU (#53201)
Signed-off-by: Yan Ma <yan.ma@intel.com>
Yan Ma committed
Aug 21, 2026 at 18:13 UTC
6feafb8b7d9c5b21ab747279182c820ef2d9265e
1 file changed
+9
vllm/model_executor/layers/rotary_embedding/mrope.py
+9
@@ -422,6 +422,15 @@ class MRotaryEmbedding(RotaryEmbeddingBase):
422
) -> tuple[torch.Tensor, torch.Tensor | None]:
423
return self.forward_native(positions, query, key, offsets)
424
425
+ def forward_xpu(
426
+ self,
427
+ positions: torch.Tensor,
428
+ query: torch.Tensor,
429
+ key: torch.Tensor | None = None,
430
+ offsets: torch.Tensor | None = None,
431
+ ) -> tuple[torch.Tensor, torch.Tensor | None]:
432
+ return self.forward_cuda(positions, query, key, offsets)
433
+
434
@staticmethod
435
def get_next_input_positions(
436
mrope_position_delta: int,