add vxpu

2026-02-05 19:36:06 +08:00
parent 070bfa4a73
commit e273ef01b8
131 changed files with 28539 additions and 2 deletions
--- a/vllm_kunlun/init.py
+++ b/vllm_kunlun/init.py
@@ -28,6 +28,15 @@ def _custom_import(module_name, globals=None, locals=None, fromlist=(), level=0)
            module = importlib.import_module(target_module)
            sys.modules[module_name] = module
            sys.modules[target_module] = module
+
+        patch_modules = {
+            "vllm.v1.engine.core": "vllm_kunlun.patch.platform.patch_core",
+            "vllm.executor.executor_base": "vllm_kunlun.patch.platform.patch_executor",
+        }
+
+        if module_name in patch_modules:
+            importlib.import_module(patch_modules[module_name])
+
    except Exception:
        pass