[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
|
${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: disabling MLAPO operators")
|
||||||
message(STATUS "310P hardware detected: excluding batch_matmul_transpose operators")
|
message(STATUS "310P hardware detected: excluding batch_matmul_transpose operators")
|
||||||
list(REMOVE_ITEM VLLM_ASCEND_CUSTOM_OP ${VLLM_ASCEND_CUSTOM_OP_EXCLUDE})
|
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}")
|
message("TORCH_NPU_PATH is ${TORCH_NPU_PATH}")
|
||||||
|
|
||||||
if(SOC_VERSION STREQUAL "ascend310p3")
|
if(SOC_VERSION MATCHES "ascend310p.*")
|
||||||
file(GLOB VLLM_ASCEND_SRC
|
file(GLOB VLLM_ASCEND_SRC
|
||||||
${CMAKE_CURRENT_SOURCE_DIR}/csrc/*.cpp
|
${CMAKE_CURRENT_SOURCE_DIR}/csrc/*.cpp
|
||||||
${CMAKE_CURRENT_SOURCE_DIR}/csrc/aclnn_torch_adapter/*.cpp)
|
${CMAKE_CURRENT_SOURCE_DIR}/csrc/aclnn_torch_adapter/*.cpp)
|
||||||
|
|||||||
Reference in New Issue
Block a user