xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-04-20 15:27:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Misc] Missing error message for custom ops import (#5282)

Browse Source
This commit is contained in:
Jie Fu (傅杰) 2024-06-07 11:17:21 +08:00 committed by GitHub
parent ccdc490dda
commit 15063741e3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
vllm/_custom_ops.py
View File

@ -5,8 +5,10 @@ import torch
try:
from vllm._C import cache_ops as vllm_cache_ops
from vllm._C import ops as vllm_ops
except ImportError:
pass
except ImportError as e:
from vllm.logger import init_logger
logger = init_logger(__name__)
logger.warning("Failed to import from vllm._C with %r", e)
# activation ops