EngineX/xc-llm-kunlun
3
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

[Bugfix] Fixed Kunlun Graph Failed (#193)

Browse Source 
Signed-off-by: dongxinyu03 <dongxinyu03@baidu.com>
This commit is contained in:
Xinyu Dong
2026-02-11 18:52:18 +08:00
committed by GitHub
parent fc48b79ae9
commit 070bfa4a73
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
vllm_kunlun/vllm_utils_wrapper.py
View File

@@ -12,6 +12,14 @@ import vllm.envs as envs
import vllm.utils as _orig import vllm.utils as _orig
from torch.library import Library, register_fake 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): def patch_annotations_for_schema(func):
"""patch_annotations_for_schema""" """patch_annotations_for_schema"""