ttanzhiqiang 2498d297ae add custom ascendc kernel vocabparallelembedding (#796) ...

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

pytest -s benchmarks/ops/ben_vocabparallelembedding.py
pytest -s tests/ops/test_vocabparallelembedding.py

---------

Signed-off-by: ttanzhiqiang <389825161@qq.com>

2025-06-12 10:44:33 +08:00

..

advance_step.cpp

[Performance]: Custom AscendC Kernel of Multi-Step Prepare Input (#814)

2025-05-20 09:31:30 +08:00

get_masked_input_and_mask_kernel.cpp

add custom ascendc kernel vocabparallelembedding (#796)

2025-06-12 10:44:33 +08:00

pos_encoding_kernels.cpp

[MISC] fix format check error (#654)

2025-04-29 11:14:19 +08:00

types.h

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

2025-04-03 14:52:34 +08:00

utils.h

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

2025-04-03 14:52:34 +08:00