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

xjpang · Jul 8, 2024 · a23a014 · a23a014
1 parent d41f79a
commit a23a014
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/vllm/_custom_ops.py b/vllm/_custom_ops.py
@@ -5,8 +5,10 @@
 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