Don't throw "Failed to import from vllm._C" warning on HPU (vllm-proj…

…ect#326)
jikunshang · Sep 24, 2024 · cf4c3e5 · cf4c3e5
1 parent 58d5cde
commit cf4c3e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vllm/_custom_ops.py b/vllm/_custom_ops.py
@@ -11,7 +11,7 @@
 
 logger = init_logger(__name__)
 
-if not current_platform.is_tpu():
+if not current_platform.is_tpu() and not current_platform.is_hpu():
     try:
         import vllm._C
     except ImportError as e: