Skip to content

Commit 6904af5

Browse files
committed
add mla_preprocess kernel
Signed-off-by: mojave2 <[email protected]>
1 parent 3d21ed9 commit 6904af5

30 files changed

+9038
-1
lines changed

CMakeLists.txt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,11 +44,13 @@ else()
4444
endif()
4545

4646
include(${ASCENDC_CMAKE_DIR}/ascendc.cmake)
47+
4748
file(GLOB KERNEL_FILES
4849
${CMAKE_CURRENT_SOURCE_DIR}/csrc/kernels/*.cpp)
4950

5051
ascendc_library(vllm_ascend_kernels SHARED
5152
${KERNEL_FILES}
53+
${CMAKE_CURRENT_SOURCE_DIR}/csrc/mla_preprocess/op_kernel/mla_preprocess_kernel.cpp
5254
)
5355

5456
message("TORCH_NPU_PATH is ${TORCH_NPU_PATH}")
@@ -90,7 +92,11 @@ target_link_libraries(
9092
libtorch_npu.so
9193
vllm_ascend_kernels
9294
ascendcl
95+
tiling_api
96+
register
9397
platform
98+
ascendalog
99+
dl
94100
)
95101

96102
target_link_options(vllm_ascend_C PRIVATE "-Wl,-rpath,$ORIGIN:$ORIGIN/lib")

0 commit comments

Comments
 (0)