[Fixbug] Fix soc_version for 310p

Signed-off-by: hfadzxy <starmoon_zhang@163.com>

Author: zhangxinyuehfad

vllm_ascend/__init__.py

@@ -15,6 +15,10 @@
 # This file is a part of the vllm-ascend project.
 #
 
+import logging
+
+logger = logging.getLogger(__name__)
+
 # Supported SOC_VERSION codes
 ASCEND_A2_SOC_VERSION = range(220, 226)
 ASCEND_A3_SOC_VERSION = range(250, 256)
@@ -28,20 +32,33 @@ def register():
 
 
 def register_model():
-    import torch_npu  # type: ignore
-
-    from vllm_ascend import _build_info  # type: ignore
-    raw = torch_npu.npu.get_soc_version()
-    soc_version = (
-        "ASCEND910B1" if raw in ASCEND_A2_SOC_VERSION else
-        "ASCEND910_9392" if raw in ASCEND_A3_SOC_VERSION else
-        "ASCEND310P3" if raw in ASCEND_310P_SOC_VERSION else "UNDEFINED")
-    if soc_version == "UNDEFINED":
-        raise RuntimeError("Unsupported or undefined Ascend SOC version.")
-    elif soc_version != _build_info.__soc_version__:
-        raise RuntimeError(
-            f"Built for SOC version {_build_info.__soc_version__}, but need running on default {soc_version}. Please reinstall vllm-ascend with the default SOC_VERSION."
-        )
+    try:
+        import torch_npu  # type: ignore
+    except ImportError:
+        logger.warning("torch_npu not found. Proceeding with CPU testing.")
     else:
-        from .models import register_model
-        register_model()
+        from vllm_ascend import _build_info  # type: ignore
+        ascend_soc_version = getattr(_build_info, "__ascend_soc_version__",
+                                     None)
+        raw = torch_npu.npu.get_soc_version()
+        soc_version = (
+            "ASCEND910B1" if raw in ASCEND_A2_SOC_VERSION else
+            "ASCEND910_9392" if raw in ASCEND_A3_SOC_VERSION else
+            "ASCEND310P3" if raw in ASCEND_310P_SOC_VERSION else "UNDEFINED")
+
+        if soc_version == "UNDEFINED":
+            raise RuntimeError("Unsupported or undefined Ascend SOC version.")
+        elif soc_version != _build_info.__soc_version__:
+            raise RuntimeError(
+                f"Built for SOC version {_build_info.__soc_version__}, "
+                f"but running on SOC version {soc_version}. "
+                f"Please reinstall vllm-ascend with 'export SOC_VERSION={soc_version}' ."
+            )
+        elif ascend_soc_version is None:
+            raise RuntimeError(
+                "vllm_ascend._build_info is missing '__ascend_soc_version__'. "
+                f"Please reinstall vllm-ascend with 'export SOC_VERSION={soc_version}' ."
+            )
+
+    from .models import register_model
+    register_model()

vllm_ascend/utils.py

@@ -22,7 +22,6 @@
 import math
 import os
 from contextlib import contextmanager, nullcontext
-from enum import Enum
 from threading import Lock
 from typing import TYPE_CHECKING, Any, List, Optional, Tuple, Union
 

vllm_ascend/worker/model_runner_v1.py

@@ -116,7 +116,7 @@
 from vllm_ascend.spec_decode.mtp_proposer import MtpProposer
 from vllm_ascend.utils import (ACL_FORMAT_FRACTAL_ND, ACL_FORMAT_FRACTAL_NZ,
                                ProfileExecuteDuration, is_310p,
-                               lmhead_tp_enable)
+                               lmhead_tp_enable, vllm_version_is)
 from vllm_ascend.worker.npu_input_batch import CachedRequestState, InputBatch
 
 if TYPE_CHECKING: