[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,31 @@ 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
+
+        from vllm_ascend import _build_info  # type: ignore
+
+        if torch_npu.npu.is_available():
+            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 need running on default {soc_version}. "
+                    "Please reinstall vllm-ascend with the default SOC_VERSION."
+                )
+
+    except Exception as e:
+        logger.warning(
+            f"SOC version check failed: {e}. Proceeding with model registration."
         )
-    else:
-        from .models import register_model
-        register_model()
+
+    from .models import register_model
+    register_model()