[310P]: fix 310p image cannot build (#6238)
Fix 310p image build error
- vLLM version: v0.14.1
- vLLM main:
d68209402d
---------
Signed-off-by: Tflowers-0129 <2906339855@qq.com>
This commit is contained in:
@@ -70,7 +70,7 @@ set(VLLM_ASCEND_CUSTOM_OP_EXCLUDE
|
||||
${CMAKE_CURRENT_SOURCE_DIR}/csrc/batch_matmul_transpose/op_kernel/batch_matmul_transpose_kernel.cpp
|
||||
)
|
||||
|
||||
if(SOC_VERSION STREQUAL "ascend310p3")
|
||||
if(SOC_VERSION MATCHES "ascend310p.*")
|
||||
message(STATUS "310P hardware detected: disabling MLAPO operators")
|
||||
message(STATUS "310P hardware detected: excluding batch_matmul_transpose operators")
|
||||
list(REMOVE_ITEM VLLM_ASCEND_CUSTOM_OP ${VLLM_ASCEND_CUSTOM_OP_EXCLUDE})
|
||||
@@ -82,7 +82,7 @@ ascendc_library(vllm_ascend_kernels SHARED
|
||||
|
||||
message("TORCH_NPU_PATH is ${TORCH_NPU_PATH}")
|
||||
|
||||
if(SOC_VERSION STREQUAL "ascend310p3")
|
||||
if(SOC_VERSION MATCHES "ascend310p.*")
|
||||
file(GLOB VLLM_ASCEND_SRC
|
||||
${CMAKE_CURRENT_SOURCE_DIR}/csrc/*.cpp
|
||||
${CMAKE_CURRENT_SOURCE_DIR}/csrc/aclnn_torch_adapter/*.cpp)
|
||||
|
||||
Reference in New Issue
Block a user