Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions vllm/model_executor/models/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -258,9 +258,9 @@ def verify_and_update_config(vllm_config: "VllmConfig") -> None:
if structured_outputs_config.reasoning_parser == "":
structured_outputs_config.reasoning_parser = "openai_gptoss"

# Increase the max capture size from 512 to 992 for performance.
# Increase the max capture size from 512 to 1024 for performance.
# NOTE(woosuk): This will increase the number of CUDA graphs
# from 67 to 81.
# from 67 to 83.
compilation_config = vllm_config.compilation_config
# Only override when the user has not set either of
# cudagraph_capture_sizes or max_cudagraph_capture_size.
Expand All @@ -269,10 +269,10 @@ def verify_and_update_config(vllm_config: "VllmConfig") -> None:
and compilation_config.max_cudagraph_capture_size is None
):
# FIXME(woosuk): When using full cuda graph with FA3, the max
# supported size is 992.
compilation_config.max_cudagraph_capture_size = 992
# supported size is 1024.
compilation_config.max_cudagraph_capture_size = 1024
logger.info(
"Overriding max cuda graph capture size to %d for performance.", 992
"Overriding max cuda graph capture size to %d for performance.", 1024
)


Expand Down