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

15063741 · Jie Fu (傅杰) · GitHub · ccdc490d · 15063741
Unverified Commit 15063741 authored Jun 07, 2024 by Jie Fu (傅杰) Committed by GitHub Jun 06, 2024
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

vllm/_custom_ops.py vllm/_custom_ops.py +4 -2

No files found.
--- a/vllm/_custom_ops.py
+++ b/vllm/_custom_ops.py
@@ -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:
+except ImportError as e:
-    pass
+    from vllm.logger import init_logger
+    logger = init_logger(__name__)
+    logger.warning("Failed to import from vllm._C with %r", e)
 # activation ops