Skip to content

Commit d900910

Browse files
Check the test3
Signed-off-by: zhihaofang1017 <zhihaofang1017@outlook.com>
1 parent 30c3e94 commit d900910

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm_ascend/ops/rotary_embedding.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -287,8 +287,8 @@ def mrope_forward(
287287
) -> Tuple[torch.Tensor, torch.Tensor]:
288288
import torch_npu
289289
mrope_section = [0, 0, 0] if positions.ndim == 1 else self.mrope_section
290-
291-
query, key = torch_npu.npu_mrope(positions.clone(),
290+
positions = positions.clone().detach().continuous()
291+
query, key = torch_npu.npu_mrope(positions,
292292
query.contiguous(),
293293
key.contiguous(),
294294
self.cos_sin_cache.contiguous(),

0 commit comments

Comments
 (0)