[AMD][DeepSeek V4] Set SGLANG_OPT_FLASHMLA_SPARSE_PREFILL to false on hip code path (#30237)
This commit is contained in:
@@ -28,8 +28,8 @@ def apply_deepseek_v4_defaults(server_args: ServerArgs, model_arch: str) -> None
|
|||||||
# currently returns incorrect output for DeepSeek-V4-Flash on ROCm/HIP
|
# currently returns incorrect output for DeepSeek-V4-Flash on ROCm/HIP
|
||||||
# (MI355X), which breaks the disaggregation nightly. Keep the previous
|
# (MI355X), which breaks the disaggregation nightly. Keep the previous
|
||||||
# (dense prefill) behavior on ROCm until the sparse kernel is validated
|
# (dense prefill) behavior on ROCm until the sparse kernel is validated
|
||||||
# there; an explicit env var still overrides this.
|
# there;
|
||||||
if is_hip() and not envs.SGLANG_OPT_FLASHMLA_SPARSE_PREFILL.is_set():
|
if is_hip():
|
||||||
logger.warning(
|
logger.warning(
|
||||||
"Disabling SGLANG_OPT_FLASHMLA_SPARSE_PREFILL by default on ROCm/HIP "
|
"Disabling SGLANG_OPT_FLASHMLA_SPARSE_PREFILL by default on ROCm/HIP "
|
||||||
f"for {model_arch}; set it explicitly to override."
|
f"for {model_arch}; set it explicitly to override."
|
||||||
|
|||||||
Reference in New Issue
Block a user