Unverified Commit 3112271f authored by Yan Ma's avatar Yan Ma Committed by GitHub
Browse files

[XPU] log clean up for XPU platform (#20553)


Signed-off-by: default avataryan <yan.ma@intel.com>
parent 1fd471e9
......@@ -13,7 +13,8 @@ from vllm.scalar_type import ScalarType
logger = init_logger(__name__)
if not current_platform.is_tpu() and not current_platform.is_hpu():
if not current_platform.is_tpu() and not current_platform.is_hpu()\
and not current_platform.is_xpu():
try:
import vllm._C
except ImportError as e:
......
......@@ -37,7 +37,7 @@ class XPUPlatform(Platform):
dtype: torch.dtype, kv_cache_dtype: Optional[str],
block_size: int, use_v1: bool,
use_mla: bool) -> str:
if selected_backend != _Backend.IPEX:
if selected_backend is not None and selected_backend != _Backend.IPEX:
logger.info("Cannot use %s backend on XPU.", selected_backend)
use_v1 = envs.VLLM_USE_V1
if not use_v1:
......@@ -133,8 +133,7 @@ class XPUPlatform(Platform):
@classmethod
def is_pin_memory_available(cls):
logger.warning("Pin memory is not supported on XPU.")
return False
return True
@classmethod
def get_current_memory_usage(cls,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment