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>
15 lines
287 B
TOML
15 lines
287 B
TOML
[build-system]
|
|
# Should be mirrored in requirements.txt
|
|
requires = [
|
|
"setuptools>=64",
|
|
"setuptools-scm>=8",
|
|
"cmake>=3.26",
|
|
"pybind11",
|
|
"decorator",
|
|
"pyyaml",
|
|
"scipy",
|
|
"torch_npu >= 2.5.1rc1",
|
|
"torch >= 2.5.1"
|
|
]
|
|
build-backend = "setuptools.build_meta"
|