diff --git a/vllm_ascend/ops/rotary_embedding.py b/vllm_ascend/ops/rotary_embedding.py index 552ef9f6..11f0f232 100644 --- a/vllm_ascend/ops/rotary_embedding.py +++ b/vllm_ascend/ops/rotary_embedding.py @@ -74,7 +74,7 @@ def rope_deepseek_forward_oot( ori_query_shape, ori_key_shape = query.shape, key.shape query = query.contiguous().view(query.shape[0], -1) key = key.contiguous().view(query.shape[0], -1) - torch_npu.npu_rope( + torch_npu._npu_rotary_embedding( positions, query, key,