Skip to content
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

[Core] Init vllm-ascend #3

Merged
merged 2 commits into from
Feb 5, 2025
Merged

[Core] Init vllm-ascend #3

merged 2 commits into from
Feb 5, 2025

Conversation

Yikun
Copy link
Collaborator

@Yikun Yikun commented Feb 5, 2025

What this PR does / why we need it?

vLLM Ascend plugin (vllm-ascend) is a backend plugin for running vLLM on the Ascend NPU.

This plugin is the recommended approach for supporting the Ascend backend within the vLLM community. It adheres to the principles outlined in the [RFC]: Hardware pluggable, providing a hardware-pluggable interface that decouples the integration of the Ascend NPU with vLLM.

This patch also include changes to make CI work and use cache speed up e2e test, including:

  1. Change push (post merge ci) and pull_request (pr ci) trigger branch to main
  2. Make mypy work by ignore base_communicator and clear unused deps
  3. Several improvements for vllm_ascend_test:
    - use cache (pip, ms, hf) speed up e2e test (25mins --> 5mins)
    - switch git clone command to action/checkout to speedup checkout and
    - Enable sv for pytest for better info dump
    - Remove network host to resole docker: conflicting ontions: cannot attach both user-defined and non-user-definednetwork-modes, which is a problem on docker 1.45 but not on 1.39.
  4. Adapt MLA decode optimizations: vllm-project/vllm@cabaf4e

Does this PR introduce any user-facing change?

Yes, init the PR.

How was this patch tested?

  • This is the first PR to make ascend NPU work on vLLM. All code is tested on ascend with vLLM V0 Engine.
  • CI passed

wangxiyuan and others added 2 commits February 5, 2025 10:29
Init vllm-ascend from POC code

Co-authored-by: wangxiyuan <wangxiyuan1007@gmail.com>
Co-authored-by: Yikun Jiang <yikunkero@gmail.com>
Co-authored-by: MengqingCao <cmq0113@163.com>
Co-authored-by: wangshuai09 <391746016@qq.com>
Co-authored-by: Shanshan Shen <467638484@qq.com>
Co-authored-by: wangli <wangli858794774@gmail.com>
Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
This patch make CI work and use cache speed up e2e test, including:
  1. Change push (post merge ci) and pull_request (pr ci) trigger branch to main
  2. Make mypy work by ignore base_communicator and clear unused deps
  3. Several improvements for vllm_ascend_test:
  - use cache (pip, ms, hf) speed up e2e test (25mins --> 5mins)
  - switch `git clone` command to `action/checkout` to speedup checkout and
  - Enable sv for pytest for better info dump
  - Remove network host to resole `docker: conflicting ontions: cannot attach both user-defined and non-user-definednetwork-modes`, which is a problem on docker 1.45 but not on 1.39.
  4. Adapt MLA decode optimizations: vllm-project/vllm@cabaf4e

Signed-off-by: Yikun Jiang <yikunkero@gmail.com>
@wangxiyuan wangxiyuan merged commit d5e7756 into vllm-project:main Feb 5, 2025
10 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants