From 418fccf0bc8957c7379cb87230f30261f02c35c8 Mon Sep 17 00:00:00 2001
From: Shaoxu Cheng <2906339855@qq.com>
Date: Mon, 26 Jan 2026 11:37:19 +0800
Subject: [PATCH] [310P]: fix 310p image cannot build (#6238)

Fix 310p image build error

- vLLM version: v0.14.1
- vLLM main:
https://github.com/vllm-project/vllm/commit/d68209402ddab3f54a09bc1f4de9a9495a283b60

---------

Signed-off-by: Tflowers-0129 <2906339855@qq.com>
---
 CMakeLists.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 00f3a9dc..9e3564e3 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -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)