Skip to content

Commit 30c3e94

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

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

vllm_ascend/ops/rotary_embedding.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -288,8 +288,7 @@ def mrope_forward(
288288
import torch_npu
289289
mrope_section = [0, 0, 0] if positions.ndim == 1 else self.mrope_section
290290

291-
positions = positions
292-
query, key = torch_npu.npu_mrope(positions,
291+
query, key = torch_npu.npu_mrope(positions.clone(),
293292
query.contiguous(),
294293
key.contiguous(),
295294
self.cos_sin_cache.contiguous(),

0 commit comments

Comments
 (0)