From 070bfa4a737b2062e4140beda5b56286bc22baa8 Mon Sep 17 00:00:00 2001
From: Xinyu Dong <dongxinyu03@baidu.com>
Date: Wed, 11 Feb 2026 18:52:18 +0800
Subject: [PATCH] [Bugfix] Fixed Kunlun Graph Failed (#193)

Signed-off-by: dongxinyu03 <dongxinyu03@baidu.com>
---
 vllm_kunlun/vllm_utils_wrapper.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/vllm_kunlun/vllm_utils_wrapper.py b/vllm_kunlun/vllm_utils_wrapper.py
index ebe9713..c0606a7 100644
--- a/vllm_kunlun/vllm_utils_wrapper.py
+++ b/vllm_kunlun/vllm_utils_wrapper.py
@@ -12,6 +12,14 @@ import vllm.envs as envs
 import vllm.utils as _orig
 from torch.library import Library, register_fake
 
+try:
+    import vllm_kunlun._kunlun  # noqa: F401
+except ImportError as e:
+    try:
+        from . import _kunlun  # noqa: F401, F403
+    except ImportError:
+        print(f"Warning: Failed to load vllm_kunlun native extension: {e}")
+
 
 def patch_annotations_for_schema(func):
     """patch_annotations_for_schema"""