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

This commit is contained in:
Jie Fu (傅杰)
2024-06-07 11:17:21 +08:00
committed by GitHub
parent ccdc490dda
commit 15063741e3

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