EngineX/xc-llm-ascend
3
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

[CI] Fix unsolved bugs caused by pta api change. (#190)

Browse Source 
This PR fix some unsolved bugs caused by pta api change.

Signed-off-by: hw_whx <wanghexiang7@huawei.com>
Co-authored-by: hw_whx <wanghexiang7@huawei.com>
This commit is contained in:
whx
2025-02-27 19:52:28 +08:00
committed by GitHub
parent 6aed83335c
commit 14bca9911a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
vllm_ascend/ops/rotary_embedding.py
View File

@@ -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,