[Bugfix] Restore balance scheduling patch for v0.17.0 (#7479)

### What this PR does / why we need it?
Restore previously introduced patches:
- https://github.com/vllm-project/vllm-ascend/pull/5212

- vLLM version: v0.17.0
- vLLM main:
8b6325758c

Signed-off-by: MrZ20 <2609716663@qq.com>
This commit is contained in:
SILONG ZENG
2026-03-19 20:12:57 +08:00
committed by GitHub
parent 95e1dc11d8
commit eb92e7d50e

View File

@@ -19,7 +19,8 @@ import os
import vllm_ascend.patch.platform.patch_distributed # noqa
import vllm_ascend.patch.platform.patch_fusion_matcher_compat_ops # noqa
import vllm_ascend.patch.platform.patch_kv_cache_interface # noqa
from vllm_ascend.utils import is_310p
from vllm_ascend import envs
from vllm_ascend.utils import is_310p, vllm_version_is
if not is_310p():
import vllm_ascend.patch.platform.patch_mamba_config # noqa
@@ -31,3 +32,5 @@ import vllm_ascend.patch.platform.patch_torch_accelerator # noqa
if os.getenv("DYNAMIC_EPLB", "false").lower() in ("true", "1") or os.getenv("EXPERT_MAP_RECORD", "false") == "true":
import vllm_ascend.patch.platform.patch_multiproc_executor # noqa
if envs.VLLM_ASCEND_BALANCE_SCHEDULING and vllm_version_is("0.17.0"):
import vllm_ascend.patch.platform.patch_balance_schedule # noqa