add mla_preprocess kernel

Signed-off-by: mojave2 <chenchen145@huawei.com>

Author: kiscad

.pre-commit-config.yaml

@@ -12,8 +12,8 @@ repos:
     - id: codespell
       args: [
         --toml, pyproject.toml,
-        '--skip', 'tests/e2e/multicard/test_torchair_graph_mode.py,tests/prompts/**,./benchmarks/sonnet.txt,*tests/lora/data/**,build/**,./vllm_ascend.egg-info/**,.github/**,typos.toml',
-        '-L', 'CANN,cann,NNAL,nnal,ASCEND,ascend,EnQue,CopyIn'
+        '--skip', 'tests/e2e/multicard/test_torchair_graph_mode.py,csrc/mla_preprocess/**,tests/prompts/**,./benchmarks/sonnet.txt,*tests/lora/data/**,build/**,./vllm_ascend.egg-info/**,.github/**,typos.toml',
+        '-L', 'CANN,cann,NNAL,nnal,ASCEND,ascend,EnQue,CopyIn,ArchType,AND'
       ]
       additional_dependencies:
         - tomli
@@ -35,6 +35,10 @@ repos:
   rev: v1.32.0
   hooks:
   - id: typos
+    args: [
+      "--force-exclude",
+      "--exclude", "csrc/mla_preprocess/**"
+    ]
 - repo: https://github.com/PyCQA/isort
   rev: 6.0.1
   hooks:

CMakeLists.txt

@@ -44,11 +44,13 @@ else()
 endif()
 
 include(${ASCENDC_CMAKE_DIR}/ascendc.cmake)
+
 file(GLOB KERNEL_FILES
 ${CMAKE_CURRENT_SOURCE_DIR}/csrc/kernels/*.cpp)
 
 ascendc_library(vllm_ascend_kernels SHARED
     ${KERNEL_FILES}
+    ${CMAKE_CURRENT_SOURCE_DIR}/csrc/mla_preprocess/op_kernel/mla_preprocess_kernel.cpp
 )
 
 message("TORCH_NPU_PATH is ${TORCH_NPU_PATH}")
@@ -90,7 +92,11 @@ target_link_libraries(
   libtorch_npu.so
   vllm_ascend_kernels
   ascendcl
+  tiling_api
+  register
   platform
+  ascendalog
+  dl
 )
 
 target_link_options(vllm_ascend_C PRIVATE "-Wl,-rpath,$ORIGIN:$ORIGIN/lib")