Skip to content

GitLab

Unverified Commit 5f5664b3 authored by Kunshang Ji's avatar Kunshang Ji Committed by GitHub
Browse files

[XPU] Fix compile size for xpu (#23069) 


Signed-off-by: default avatarKunshang Ji <kunshang.ji@intel.com>
parent 89657a55
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
vllm/config/__init__.py
View file @ 5f5664b3
...@@ -3548,7 +3548,7 @@ class VllmConfig: ...@@ -3548,7 +3548,7 @@ class VllmConfig:
if self.compilation_config.pass_config.enable_sequence_parallelism: if self.compilation_config.pass_config.enable_sequence_parallelism:
self.compilation_config.custom_ops.append("+rms_norm") self.compilation_config.custom_ops.append("+rms_norm")
if current_platform.is_cuda_alike(): if current_platform.is_cuda_alike() or current_platform.is_xpu():
# if cudagraph_mode is not explicitly set by users, set default # if cudagraph_mode is not explicitly set by users, set default
# value # value
if self.compilation_config.cudagraph_mode is None: if self.compilation_config.cudagraph_mode is None:
......
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