Skip to content

[NPU] add A5 support for A2E and E2A ops - #276

Open
wangyibo1005 wants to merge 1 commit into
vllm-project:mainfrom
wangyibo1005:main
Open

[NPU] add A5 support for A2E and E2A ops#276
wangyibo1005 wants to merge 1 commit into
vllm-project:mainfrom
wangyibo1005:main

Conversation

@wangyibo1005

Copy link
Copy Markdown

A5 HCCL rejects the 910C MC2 tiling/window ABI, so register CANN 9.1 ascend950 and the CANN 8.5 ascend910_95 alias, use the MTE comm engine, and read A5 window addresses without changing the 910C path.

PLEASE FILL IN THE PR DESCRIPTION AND MAKE SURE THE CHECKLIST ITEMS HAVE BEEN CONSIDERED.

Purpose

Issue

  • Related issue(s): #
  • Closing keyword, only if fully resolved: Closes #

Scope

  • In scope:
  • Out of scope:

Implementation Notes

Test Plan

Test Result

Docs Impact

  • Files updated:
  • If none, reason:

Essential PR Checklist
  • Purpose is clear and linked to public context when possible.
  • Scope is bounded.
  • Compatibility with vLLM v0.26.0 is considered.
  • No changes are made to the vLLM source checkout.
  • Plugin-owned classes or explicit dotted class paths are preferred over monkey patches.
  • Any compat shim or monkey patch is isolated, idempotent, version-guarded, documented, and tested.
  • Imports remain CPU-safe; CUDA-heavy work is delayed or GPU-gated.
  • Validation evidence is included, including skipped GPU tests when applicable.
  • Documentation impact is stated.

A5 HCCL rejects the 910C MC2 tiling/window ABI, so register CANN 9.1
ascend950 and the CANN 8.5 ascend910_95 alias, use the MTE comm engine,
and read A5 window addresses without changing the 910C path.

Signed-off-by: wangyibo1005 <2633333316@qq.com>

Copy link
Copy Markdown
Collaborator

Thanks for adding A5 support! Has this change also been validated on A3? Could you please run the relevant A3 test cases as well?

jiangkuaixue123 added a commit that referenced this pull request Aug 27, 2026
## Summary

- configure the repository-level typos allowlist in `pyproject.toml`
- preserve `CANN`/`cann` and `TBE`/`tbe` when checked as plain words
- preserve Ascend and vendored-code identifiers, including the `ND`
data-format suffix
- prevent unsafe autofixes such as `CANN_HOME` to `CAN_HOME`, `tbe` to
`the`, and `FORMAT_ND` to `FORMAT_AND`

## Validation

- confirmed `CANN`, `CANN_HOME`, `_cann_ops_custom`, `tbe`, and
`ge::FORMAT_ND` are accepted
- confirmed the actual typo `DATA_FULSH` is still reported as
`DATA_FLUSH`

Related: #276

Signed-off-by: jiangkuaixue123 <jiangxiaozhou111@163.com>
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