[core] Support custom ascendc kernels in vllm-ascend (#233)

This PR add custom ascendc kernel rotary_embedding support in
vllm-ascend, related CMakeLists and setuptools is also added in this PR.

Related: https://github.com/vllm-project/vllm-ascend/issues/156

---------

Signed-off-by: ganyi <pleaplusone.gy@gmail.com>
This commit is contained in:
Pleaplusone
2025-04-03 14:52:34 +08:00
committed by GitHub
parent 14d9a64047
commit ce8259975e
15 changed files with 1378 additions and 9 deletions

View File

@@ -3,9 +3,12 @@
requires = [
"setuptools>=64",
"setuptools-scm>=8",
"cmake>=3.26",
"pybind11",
"decorator",
"pyyaml",
"scipy",
"torch-npu >= 2.5.1rc1"
"torch_npu >= 2.5.1rc1",
"torch >= 2.5.1"
]
build-backend = "setuptools.build_meta"