Skip to content

GitLab

Unverified Commit 6951fcd4 authored by Huanxing's avatar Huanxing Committed by GitHub
Browse files

[XPU] Automatically detect target platform as XPU in build. (#37634) 


Signed-off-by: default avatarhuanxing <huanxing.shen@intel.com>
parent 39474513
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
setup.py
View file @ 6951fcd4
......@@ -54,6 +54,9 @@ elif sys.platform.startswith("linux") and os.getenv("VLLM_TARGET_DEVICE") is Non
if torch.version.hip is not None:
VLLM_TARGET_DEVICE = "rocm"
logger.info("Auto-detected ROCm")
elif torch.version.xpu is not None:
VLLM_TARGET_DEVICE = "xpu"
logger.info("Auto-detected XPU")
elif torch.version.cuda is not None:
VLLM_TARGET_DEVICE = "cuda"
logger.info("Auto-detected CUDA")
......
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