Skip to content

Commit ca400b8

Browse files
committed
raise error when get graph pool on XPU
Signed-off-by: Kunshang Ji <[email protected]>
1 parent 937d325 commit ca400b8

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

vllm/platforms/xpu.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -96,8 +96,8 @@ def check_and_update_config(cls, vllm_config: VllmConfig) -> None:
9696
if compilation_config.cudagraph_mode is None or \
9797
compilation_config.cudagraph_mode.max_cudagraph_mode() \
9898
!= CUDAGraphMode.NONE:
99-
logger.info("[XPU] CUDA graph is not supported on XPU, "
100-
"disabling cudagraphs.")
99+
logger.info("[XPU] CUDA graph is not supported on XPU, disabling "
100+
"cudagraphs. Fallback to cudagraph_mode=NONE")
101101
compilation_config.cudagraph_mode = CUDAGraphMode.NONE
102102

103103
# check and update parallel config
@@ -180,6 +180,4 @@ def get_global_graph_pool(self) -> Any:
180180
"""
181181
Currently xpu does NOT support Graph model.
182182
"""
183-
logger.warning("XPU does not support graph pool. This may be unsafe "
184-
"if you enable graphmode on XPU currently.")
185-
return None
183+
raise NotImplementedError("XPU does not support Graph model.")

0 commit comments

Comments
 (0)