Refactor: extract _init_vendor_device for vendor-specific early patches

Author: luoyc123

vllm_fl/__init__.py

@@ -94,10 +94,17 @@ def _patch_custom_ops():
     register_op_schemas()
 
 
+def _init_vendor_device():
+    """Vendor-specific device initialization patches."""
+    from vllm_fl.utils import DeviceInfo
+    if DeviceInfo().vendor_name == "kunlunxin":
+        from vllm_fl.dispatch.backends.vendor.kunlunxin.patches.patch_fla_utils import _patch_xpu_get_device
+        _patch_xpu_get_device()
+
+
 def register():
     """Register the FL platform."""
-    from vllm_fl.dispatch.backends.vendor.kunlunxin.patches.patch_fla_utils import _patch_xpu_get_device
-    _patch_xpu_get_device()
+    _init_vendor_device()
 
     _patch_custom_ops()
     _patch_flash_attn_import()