Skip to content

[Fix] Enable npu_mrope by environment variables #1231

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed

Conversation

Pr0Wh1teGivee
Copy link
Contributor

@Pr0Wh1teGivee Pr0Wh1teGivee commented Jun 15, 2025

What this PR does / why we need it?

npu_rotary_embedding will trigger gatherv3's error, enable npu_mrope to solve this issue.

Does this PR introduce any user-facing change?

export VLLM_ASCEND_ENABLE_NPU_MROPE=1 to enable npu_mrope

How was this patch tested?

math500 tested in deepseek 671b

@Yikun
Copy link
Collaborator

Yikun commented Jun 15, 2025

See also: #1195

@Pr0Wh1teGivee
Copy link
Contributor Author

See also: #1195

  1. npu_mrope is available since torch_npu 25.rc1
  2. we need to merge it into 0.9.1-dev apart from main branch

@Pr0Wh1teGivee Pr0Wh1teGivee force-pushed the v0.9.1-dev branch 3 times, most recently from c65f16e to 1cdaf17 Compare June 16, 2025 07:05
@Pr0Wh1teGivee Pr0Wh1teGivee changed the title fix gatherv3 error by using npu_mrope instead enable npu_mrope by environment variables Jun 16, 2025
@Pr0Wh1teGivee Pr0Wh1teGivee changed the title enable npu_mrope by environment variables Enable npu_mrope by environment variables Jun 16, 2025
@Pr0Wh1teGivee Pr0Wh1teGivee changed the title Enable npu_mrope by environment variables [Fix] enable npu_mrope by environment variables Jun 16, 2025
@Pr0Wh1teGivee Pr0Wh1teGivee changed the title [Fix] enable npu_mrope by environment variables [Fix] Enable npu_mrope by environment variables Jun 16, 2025
Signed-off-by: Pr0Wh1teGivee <calvin_zhu0210@outlook.com>
@MengqingCao
Copy link
Collaborator

LGTM, please remember to add ut for this

Copy link

This pull request has conflicts, please resolve those before we can evaluate the pull request.

@Pr0Wh1teGivee Pr0Wh1teGivee closed this by deleting the head repository Jun 19, 2025
@ApsarasX
Copy link
Collaborator

Even if using npu_mrope, there is still a chance of encountering the gatherv3's error.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants