From 14bca9911a265bb3c75708dbd4fcdfe56d267db4 Mon Sep 17 00:00:00 2001 From: whx <56632993+whx-sjtu@users.noreply.github.com> Date: Thu, 27 Feb 2025 19:52:28 +0800 Subject: [PATCH] [CI] Fix unsolved bugs caused by pta api change. (#190) This PR fix some unsolved bugs caused by pta api change. Signed-off-by: hw_whx Co-authored-by: hw_whx --- vllm_ascend/ops/rotary_embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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,